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

Before Change

    print("LEARNING RATE:", args.lr)
    print("STYLE IMAGE:", args.style_image)
    print("CONTENT WEIGHT:", args.content_weight)
    print("STYLE WEIGHT:", args.style_weight)
    print("DATASET:", args.dataset)
    print("SAVE-MODEL DIRECTORY:", args.save_model_dir)
    print("STYLE SIZE:", args.style_size)
    print("=====================\n")

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