INNER CODE UNIT · Python

train_one_ep

FoundationVision/UniTok · main.py:77

def train_one_ep(
    args,
    data,
    epoch,
    trainer,
    start_iter,
    unitok_scheduler,
    disc_scheduler,
    visualizer,
    tokenizer
):
    dataloader = data['train'].dataloader
    num_iters = data['train'].num_batches

    metric_logger = MetricLogger(cur_epoch=epoch, total_epoch=args.epoch, delimiter='  ')
    [metric_logger.add_meter(x, SmoothedValue(window_size=1, fmt='{value:.2g}')) for x in ('glr', 'dlr')]
    [metric_logger.add_meter(x, SmoothedValue(window_size=1, fmt='{median:.2f}')) for x in ('gnm', 'dnm')]
    [metric_logger.add_meter(x, SmoothedValue(fmt='{median:.3f}')) for x in ('L1', 'Lnll', 'Ld', 'Lc', 'Wg')]

View source record →