sfaira.data.store.batch_schedule.BatchDesignFull

class sfaira.data.store.batch_schedule.BatchDesignFull(retrieval_batch_size: int, randomized_batch_access: bool, random_access: bool, **kwargs)

Emits full dataset as a single batch in each query.

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