数据加载器 tqdm 的最佳方式

Om *_*uke 4 python tqdm pytorch

如何将 tqdm 用于 data_loader ?

这是正确的方法吗?

for i,j in enumerate(data_loader,total = 100):
           pass
Run Code Online (Sandbox Code Playgroud)

Bit*_*sky 10

您需要用 tqdm 包装可迭代对象,因为他们的文档清楚地说明:

立即让您的循环显示一个智能进度表 - 只需用 tqdm(iterable) 包装任何可迭代对象,就完成了!

如果您要枚举可迭代对象,则可以执行以下操作。睡眠只是为了观想它。

from tqdm import tqdm
from time import sleep

data_loader = list(range(1000))

for i, j in enumerate(tqdm(data_loader)):
    sleep(0.01)
Run Code Online (Sandbox Code Playgroud)