14c0cf773f514788aca6b935298fa186890eecc1,test.py,,,#,16

Before Change

        target = Variable(ToTensor()(target))
        if torch.cuda.is_available():
            image = image.unsqueeze(0).cuda()
            target = target.unsqueeze().cuda()

        out = model(image)
        mse = ((target - out) ** 2).mean()

After Change

        target = Image.open(target_path + image_name)
        target = Variable(ToTensor()(target))
        if torch.cuda.is_available():
            image = image.cuda()
            target = target.cuda()

        out = model(image.unsqueeze(0))[0]
        mse = ((target - out) ** 2).mean()

Instances