hydrax.tqdm.tbatches

hydrax.tqdm.tbatches(dataloader: Dataloader[D], report_interval: int = 0, description: str = 'train', **kwargs) → Iterable[Batch[D]]

Wraps a Dataloader with tqdm progress bars, yielding each Batch.

Parameters:

• dataloader – The Dataloader to wrap. It must be deterministic, and not have been started via with.

• report_interval – Interval, in batches, to check for issues and report. 0 by default, which means reporting is disabled.

• description – A description for the progress bar. Defaults to “train”.

• kwargs – Additional keyword arguments passed to tqdm.

Returns:

An iterator over each Batch produced by the Dataloader.