1f8de025d53aed3297a588fcfe7be9fc1727ded9,neural_style/neural_style.py,,train,#Any#,19

Before Change

    else:
        kwargs = {}

    print("=====================")
    print("CURRENT TIME:", time.ctime())
    print("PYTHON VERSION:", sys.version)
    print("PYTORCH VERSION:", torch.__version__)
    print("BATCH SIZE:", args.batch_size)

After Change

    if args.cuda and not torch.cuda.is_available():
        print("ERROR: cuda is not available, try running on CPU")
        sys.exit(1)

    if args.cuda:
        torch.cuda.manual_seed(args.seed)
        kwargs = {"num_workers": 0, "pin_memory": False}

Instances