sfaira.data.store.batch_schedule.BatchDesignBalanced¶

class sfaira.data.store.batch_schedule.BatchDesignBalanced(grouping, group_weights: dict, randomized_batch_access: bool, random_access: bool, **kwargs)¶

Balanced batches across meta data partitions of data.

Attributes

`batchsplits`	Batch size of the yieleded batches.
`design`	Yields index objects for one epoch of all data.
`idx`	Protects property from uncontrolled changing.
`n_batches`

Methods