tqdm进度条管理

1. 安装

pip install tqdm

2. 一般的进度条管理

import tqdm
#for index,（data,targets） in enumerate(train_loader):
for data,targets in tqdm(train_loader):
    pass

若只输出一行可设置tqdm(train_loader,leave =False):

3. 改变输出内容

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)