05b2b1c02a8ce5f471877243ebebe5df876ccc37,models/unet/train_unet.py,,,#,209

Before Change

    parser.add_argument("--mode", choices=["train", "test"], default="train")
    parser.add_argument("--num-epochs", type=int, default=50, help="Number of training epochs")
    parser.add_argument("--gpus", type=int, default=1)
    parser.add_argument("--exp-dir", type=pathlib.Path, default="experiments",
                        help="Path where model and results should be saved")
    parser.add_argument("--exp", type=str, help="Name of the experiment")
    parser.add_argument("--checkpoint", type=pathlib.Path,
                        help="Path to pre-trained model. Use with --mode test")
    parser.add_argument("--resume", action="store_true",

After Change

    torch.manual_seed(args.seed)
    run(args)

if __name__ == "__main__":
    main()

Instances