用于数据加载器的 tqdm 的最佳方式

Question

如何将 tqdm 用于 data_loader ？这是正确的方法吗？for i,j in enumerate(data_loader,total = 100):            pass

Answer 1

你需要用 tqdm 包装 iterable，因为他们的文档清楚地说：

立即让你的循环显示一个智能进度表——只需用 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)

Answer 2

如果你想使用enumeratewith tqdm，你可以这样使用它：

for i,data in enumerate(tqdm(train_dataloader)):
    images, labels = data 
    images, labels = images.to(device), labels.to(device)
    ....