tqdm进度条 在pytorch中的使用

引入该模块：

from tqdm import tqdm

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

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

效果：

tqdm用在dataloader上其实是对每个batch和batch总数做的进度条，我也是在训练比较慢时写出来看看训练到哪了。。。