65ad632de368062843ba507a3fe740fc5015fa83,train.py,,train,#,52

Before Change

                six.print_("    * best score on validation dataset: PSNR %f dB"
                           % valid_score)
                best_model = model.copy().to_cpu()
                epoch_name = model_name.rstrip(".npz") + "_epoch%d.npz" % epoch
                chainer.serializers.save_npz(epoch_name, best_model)
            if best_count >= args.lr_decay_interval:
                best_count = 0

After Change

                          % (args.noise_level, args.color))
    else:
        model_name = args.model_name.rstrip(".npz")
    if not os.path.exists("epoch"):
        os.makedirs("epoch")

    model = srcnn.archs[args.arch](ch)
    if args.finetune is not None:
        chainer.serializers.load_npz(args.finetune, model)

Instances