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

Question

如何将 tqdm 用于 data_loader ？

这是正确的方法吗？

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

Answer 1

你需要用 tqdm 包装 iterable，因为他们的文档清楚地说：立即让你的循环显示一个智能进度表——只需用 tqdm(iterable) 包装任何可迭代对象，你就完成了！如果您正在枚举可迭代对象，则可以执行类似以下的操作。睡眠只是为了观想它。from tqdm import tqdmfrom time import sleepdata_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)     ....