INNER CODE UNIT · Python

main

NVIDIA/DeepLearningExamples · PyTorch/Classification/ConvNets/main.py:611

def main(args, model_args, model_arch):
    exp_start_time = time.time()

    (
        trainer,
        lr_policy,
        train_loader,
        train_loader_len,
        val_loader,
        logger,
        start_epoch,
        best_prec1,
    ) = prepare_for_training(args, model_args, model_arch)

    train_loop(
        trainer,
        lr_policy,
        train_loader,

View source record →