如何將 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個贊
如果你想使用enumerate
with tqdm
,你可以這樣使用它:
for i,data in enumerate(tqdm(train_dataloader)): images, labels = data images, labels = images.to(device), labels.to(device) ....
添加回答
舉報
0/150
提交
取消