tqdm的多行显示与单行显示

引入

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() # 如果单行显示这行没有

循环没明白的再看这里