Show epoch progress with Iterable datasets

[marcmk6]

Hi,

I'm wondering if there's a way to show the training progress over each epoch with torch.utils.data.IterableDataset.
The current visualization is like:

Epoch 0: : 1194it [04:50,  4.11it/s, loss=10.4, v_num=ght_]

My IterableDatasets iterate over some large files that don't fit in memory, but their lengths are known.

Thanks