时间戳

首页 > 解决方案 > 数据加载器的最佳 tqdm 方式

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

问题描述

如何将 tqdm 用于 data_loader ？

这是正确的方法吗？

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

标签： pythonpytorchtqdm

解决方案

您需要用 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)

推荐阅读