引入
from tqdm import tqdm
pbar = tqdm(total=n_batches)
单行显示
缺点,看不到变化的过程,因为会覆盖掉;优点,节省屏幕,看的舒心
for 循环取数据过程
pbar.set_description('Train loss: %.3f / loss %.3f' % (train_loss / (i + 1), loss.item()))
pbar.update()
pbar.close()
好比
多行显示
不关闭close()就是多行显示
for 循环取数据过程
pbar.set_description('Train loss: %.3f / loss %.3f' % (train_loss / (i + 1), loss.item()))
pbar.update()
补充完整循环
上面的循环就是train里面取每个数据来训练的循环,常用是这样
for i,(img, label) in enumerate(train_loader):
pbar.set_description('Train loss: %.3f / loss %.3f' % (train_loss / (i + 1), loss.item()))
pbar.update()
pbar.close() # 如果单行显示这行没有