coopihczoo.teaching.scripts_to_sort.behavioral_cloning_original.EpochOrBatchIteratorWithProgress¶

class coopihczoo.teaching.scripts_to_sort.behavioral_cloning_original.EpochOrBatchIteratorWithProgress(data_loader, n_epochs: Optional[int] = None, n_batches: Optional[int] = None, on_epoch_end: Optional[Callable[[], None]] = None, on_batch_end: Optional[Callable[[], None]] = None, progress_bar_visible: bool = True)[source]¶

Bases: object

Wraps DataLoader so that all BC batches can be processed in one for-loop.

Also uses tqdm to show progress in stdout.

Methods