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

Before Change

    print("CONTENT WEIGHT:", args.content_weight)
    print("STYLE WEIGHT:", args.style_weight)
    print("DATASET:", args.dataset)
    print("CHECKPOINT DIR:", args.checkpoint_dir)
    print("VALIDATION:", args.validation)
    print("VAL DIR:", args.val_dir)
    print("STYLE SIZE:", args.style_size)
    print("=====================\n")

After Change

    // save model
    transformer.eval()
    transformer.cpu()
    save_model_filename = "epoch_" + str(args.epochs) + "_" + str(time.ctime()).replace(" ", "_") + "_" + str(
        args.content_weight) + "_" + str(args.style_weight) + ".model"
    save_model_path = os.path.join(args.save_model_dir, save_model_filename)
    torch.save(transformer, save_model_path)

    print("\nDone :)")

Instances