torchtnt.utils.progress.estimated_steps_in_loop¶

torchtnt.utils.progress.estimated_steps_in_loop(dataloader: Iterable[object], *, max_steps: Optional[int], max_steps_per_epoch: Optional[int], epochs: Optional[int]) → Optional[int]¶

Estimate the total number of steps for the current loop.

A return value of None indicates that the number of steps couldn’t be estimated.