b0e42547fb9d10146e0aea48ac6c48504c19529d,test_single.py,,,#,10

Before Change

IMAGE_NAME = opt.image_name
MODEL_NAME = opt.model_name

model = Generator(UPSCALE_FACTOR)
if torch.cuda.is_available():
    model = model.cuda()
model.load_state_dict(torch.load("epochs/" + MODEL_NAME))

After Change

IMAGE_NAME = opt.image_name
MODEL_NAME = opt.model_name

model = Generator(UPSCALE_FACTOR).eval()
if torch.cuda.is_available():
    model = model.cuda()
model.load_state_dict(torch.load("epochs/" + MODEL_NAME))

Instances