35fdba3c5cbc7492b63a03117e2e6020350b377c,models/unet/train_unet.py,,main,#,180

Before Change

        val_check_interval=1.,
        early_stop_callback=False
    )
    if args.mode == "train":
        trainer.fit(model)
    else:  // args.mode == "test"
        trainer.test(model)


if __name__ == "__main__":
    parser = Args()
    parser.add_argument("--mode", choices=["train", "test"], default="train")
    parser.add_argument("--num-epochs", type=int, default=50, help="Number of training epochs")

After Change

        model = UnetMRIModel.load_from_checkpoint(str(args.checkpoint))
        model.hparams.sample_rate = 1.
        trainer = create_trainer(args, logger=False)
        trainer.test(model)


if __name__ == "__main__":
    parser = Args()

Instances