torchtnt.utils.tqdm.create_progress_bar

torchtnt.utils.tqdm.create_progress_bar(dataloader: Iterable[object], *, desc: str, num_epochs_completed: int, num_steps_completed: int, max_steps: Optional[int], max_steps_per_epoch: Optional[int]) → tqdm_asyncio

Constructs a tqdm() progress bar. The number of steps in an epoch is inferred from the dataloader, num_steps_completed, max_steps and max_steps_per_epoch.

Parameters:

• dataloader – an iterable of data, used to infer number of steps in an epoch.
• desc – a description for the progress bar.
• num_epochs_completed – an integer for the number of epochs completed so far int he loop.
• num_steps_completed – an integer for the number of steps completed so far in the loop.
• max_steps – an optional integer for the number of max steps in the loop.
• max_steps_per_epoch – an optional integer for the number of max steps per epoch.