01f334dcd78e262f9a0483b9b4d1a8586c1680c3,main.py,,main,#,8

Before Change

    //         config.dataset, config.batch_size, config.shuffle, 
    //         **kwargs)

    data_loader = None, None

    // instantiate trainer
    trainer = Trainer(config, data_loader)

After Change

    // either train
    if config.is_train:
        save_config(config)
        trainer.train()

    // or load a pretrained model and test
    else:
        trainer.test()

Instances