Hello, I wanted to know about loading the saved model in the enhance notebook.
m = to_gpu(SrResnet(64, scale))
m = nn.DataParallel(m, )
learn = Learner(md, SingleModel(m), opt_fn=optim.Adam)
Is this the correct way?
Getting the following error!
KeyError: ‘unexpected key “features.0.0.weight” in state_dict’
I will be very grateful for any help!