如何将 tqdm 用于 data_loader ？这是正确的方法吗？for i,j in enumerate(data_loader,total = 100):
pass
2 回答
小怪兽爱吃肉
TA贡献1852条经验 获得超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)
拉风的咖菲猫
TA贡献1995条经验 获得超2个赞
如果你想使用enumeratewith tqdm，你可以这样使用它：
for i,data in enumerate(tqdm(train_dataloader)): images, labels = data images, labels = images.to(device), labels.to(device) ....
添加回答
举报
0/150
提交
取消