pip install tqdm
import tqdm
#for index,(data,targets) in enumerate(train_loader):
for data,targets in tqdm(train_loader):
pass
若只输出一行可设置tqdm(train_loader,leave =False):
loop = tqdm(enumerate(train_loader), total =len(train_loader))
for index,(x, target)in loop:
......
#更新信息
loop.set_description(f‘Epoch [{epoch}/{num_epoch}]‘)
loop.set_postfix(loss = loss.item(),acc = running_train_acc)
原文:https://www.cnblogs.com/emmm/p/14888973.html