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)
posted @ 2021-06-16 13:11  秋夜花开  阅读(691)  评论(0编辑  收藏  举报