python - 数据加载器 tqdm 的最佳方式

Question

如何将 tqdm 用于 data_loader ？
这是正确的方法吗？

for i,j in enumerate(data_loader,total = 100):
           pass

Answer 1

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

Instantly make your loops show a smart progress meter - just wrap anyiterable with tqdm(iterable), and you’re done!

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

from tqdm import tqdm
from time import sleep

data_loader = list(range(1000))

for i, j in enumerate(tqdm(data_loader)):
    sleep(0.01)