通过tqdm实现进度条打印训练过程信息

BILLY BILLY

于 2024-08-20 17:03:41 发布

阅读量207

点赞数 1

分类专栏： torch基础文章标签： pytorch

本文链接：https://blog.csdn.net/m0_51579041/article/details/141363097

版权

torch基础专栏收录该内容

13 篇文章 0 订阅

订阅专栏

tqdm使用方法

from tqdm import tqdm

def fit_one_epoch(...):
    train_loop = tqdm(train_loader, desc='Train')
    for xx in enumerate(train_loop):
        ...
        train_loop.set_description(f'Epoch [{epoch + 1}/{config.epochs}]')
        train_loop.set_postfix({'loss_bev_seg_emb': '{0:1.2f}'.format(loss_bev_seg_emb.item()),
                                    "loss_y_offst": '{0:1.2f}'.format(loss_Y_Offset.item()),
                                    "loss_Z": '{0:1.2f}'.format(loss_Z.item()),
                                    "f1_bev_seg": '{0:1.2f}'.format(f1_ipm_seg.item()),
                                    })