tqdm進度條 在pytorch中的使用

引入該模塊：

from tqdm import tqdm

tqdm可用於所有可迭代對象，所以可以直接用在dataloader上，如下：

for data, target in tqdm(train_loader):
    ...

效果：

tqdm用在dataloader上其實是對每個batch和batch總數做的進度條，我也是在訓練比較慢時寫出來看看訓練到哪了。。。