INNER CODE UNIT · Python

do_train

OpenMeshLab/MeshXL · engine.py:32

def do_train(
    args,
    model,
    accelerator,
    optimizer,
    dataloaders,
    best_val_metrics,
    logger
):
    
    if accelerator.is_main_process:
        logger.log_messages(f"call with args: {args}")
        logger.log_messages(f"{model}")
    
    curr_iter = args.start_epoch * len(dataloaders['train'])
    max_iters = args.max_epoch * len(dataloaders['train'])
    
    time_delta = SmoothedValue(window_size=10)

View source record →