torchtnt.utils.progress.estimated_steps_in_fit

torchtnt.utils.progress.estimated_steps_in_fit(*, train_dataloader: Iterable[object], eval_dataloader: Iterable[object], epochs: Optional[int], max_steps: Optional[int], max_train_steps_per_epoch: Optional[int], max_eval_steps_per_epoch: Optional[int], eval_every_n_steps: Optional[int], eval_every_n_epochs: Optional[int]) → Optional[int]

Estimate the total number of steps for fit run.

If the number of training/eval steps couldn’t be calculated, None is returned.