如何將 tqdm 用于 data_loader ？這是正確的方法嗎？for i,j in enumerate(data_loader,total = 100):
pass
2 回答
小怪獸愛(ài)吃肉
TA貢獻(xiàn)1852條經(jīng)驗(yàn) 獲得超1個(gè)贊
你需要用 tqdm 包裝 iterable，因?yàn)樗麄兊奈臋n清楚地說(shuō)：
立即讓你的循環(huán)顯示一個(gè)智能進(jìn)度表——只需用 tqdm(iterable) 包裝任何可迭代對(duì)象，你就完成了！
如果您正在枚舉可迭代對(duì)象，則可以執(zhí)行類似以下的操作。睡眠只是為了觀想它。
from tqdm import tqdm
from time import sleep
data_loader = list(range(1000))
for i, j in enumerate(tqdm(data_loader)):
sleep(0.01)
拉風(fēng)的咖菲貓
TA貢獻(xiàn)1995條經(jīng)驗(yàn) 獲得超2個(gè)贊
如果你想使用enumeratewith tqdm，你可以這樣使用它：
for i,data in enumerate(tqdm(train_dataloader)): images, labels = data images, labels = images.to(device), labels.to(device) ....
添加回答
舉報(bào)
0/150
提交
取消