b1657087e4039232e79e2c80dadba70c9c5006c9,test.py,,,#,73

Before Change

    if args.resume:
        config = torch.load(args.resume)["config"]
    if args.device:
        os.environ["CUDA_VISIBLE_DEVICES"] = args.device

    main(config, args.resume)

After Change

                        help="indices of GPUs to enable (default: all)")

    args = parser.parse_args()
    config = ConfigParser(args)
    main(config, args.resume)

Instances