peal.data.dataloaders.DataIterator

class peal.data.dataloaders.DataIterator[source]

Bases: object

Iterator over a DataloaderMixer yielding a fixed number of batches.

One pass yields dataloader.train_config.steps_per_epoch batches obtained from dataloader.sample(), independent of the underlying dataset sizes.

Parameters:

dataloader (DataloaderMixer) – The mixer to draw batches from.

Methods

__init__(dataloader)[source]

Store the mixer and start the step counter at 0.