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()),
                                    })

输出结果