4630c3c7d27df63d3ebc9e1a527476abbbf598df,waifu2x.py,,load_models,#Any#,73

Before Change

                      % (args.noise_level, args.color))
        model_path = os.path.join(model_dir, model_name)
        if os.path.exists(model_path):
            models["noise_scale"] = srcnn.archs[args.arch](ch)
            chainer.serializers.load_npz(model_path, models["noise_scale"])
        else:
            flag = True
    if args.method == "scale" or flag:
        model_name = "anime_style_scale_%s.npz" % args.color

After Change

                      % (args.noise_level, args.color))
        model_path = os.path.join(model_dir, model_name)
        if os.path.exists(model_path):
            models["noise_scale"] = srcnn.archs[args.arch](ch)
            chainer.serializers.load_npz(model_path, models["noise_scale"])
            alpha_model_name = "anime_style_scale_%s.npz" % args.color
            alpha_model_path = os.path.join(model_dir, alpha_model_name)
            models["alpha"] = srcnn.archs[args.arch](ch)
            chainer.serializers.load_npz(alpha_model_path, models["alpha"])

Instances