6fc77d75bdb46feffa39ad2c389def87643bdcf3,test.py,,,#,49

Before Change

    args = parse_args()

    checkpoint = torch.load(args.checkpoint)
    cls_model = getattr(prosr.models, checkpoint["class_name"])

    model = cls_model(**checkpoint["params"]["G"])
    model.load_state_dict(checkpoint["state_dict"])

After Change

            fn = osp.join(args.output_dir, osp.basename(data["input_fn"][0]))
            io.imsave(fn,sr_img)
            if iid+1 == len(data_loader):
                info("Saved image in: {}".format(fn),bold=True)

        if len(args.target):
            psnr_mean /= len(dataset)
            ssim_mean /= len(dataset)
            print_evaluation("average", psnr_mean, ssim_mean)

Instances