3b825a18923f298067ac8ece781d01e83ab84016,neural_style/neural_style.py,,train,#,19

Before Change

    // save model
    transformer.eval()
    transformer.cpu()
    torch.save(transformer, args.checkpoint_dir + "/epoch_" + str(args.epochs) + "_"
               + str(time.ctime()).replace(" ", "_") + "_"
               + str(args.content_weight) + "_" + str(args.style_weight)
               + ".model")

    print("\nDone :)")

After Change

    np.random.seed(args.seed)
    torch.manual_seed(args.seed)

    if args.cuda and not torch.cuda.is_available():
        print("WARNING: torch.cuda not available, using CPU.")
        args.cuda = 0

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

Instances