LinearDynamicPoolingBatcher.Config

Component: LinearDynamicPoolingBatcher

class LinearDynamicPoolingBatcher.Config

Bases: DynamicPoolingBatcher.Config

All Attributes (including base classes)

train_batch_size: int = 16

eval_batch_size: int = 16

test_batch_size: int = 16

pool_num_batches: int = 1000

num_shuffled_pools: int = 1

scheduler_config: BatcherSchedulerConfig = BatcherSchedulerConfig()

Default JSON

{
    "train_batch_size": 16,
    "eval_batch_size": 16,
    "test_batch_size": 16,
    "pool_num_batches": 1000,
    "num_shuffled_pools": 1,
    "scheduler_config": {
        "load_path": null,
        "save_path": null,
        "freeze": false,
        "shared_module_key": null,
        "start_batch_size": 32,
        "end_batch_size": 256,
        "epoch_period": 10,
        "step_size": 1
    }
}