PoolingBatcher.ConfigΒΆ

Component: PoolingBatcher

class PoolingBatcher.Config[source]

Bases: Batcher.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
Size of a pool expressed in number of batches
num_shuffled_pools: int = 1
How many pool-sized chunks to load at a time for shuffling
Subclasses
  • DynamicPoolingBatcher.Config
  • ExponentialDynamicPoolingBatcher.Config
  • LinearDynamicPoolingBatcher.Config

Default JSON

{
    "train_batch_size": 16,
    "eval_batch_size": 16,
    "test_batch_size": 16,
    "pool_num_batches": 1000,
    "num_shuffled_pools": 1
}