Sorry to reply you so late.There are too much work for me to do as a phd candidate. Your suggestion was really help for me.But since the variable is corrected_im, should the code be as follow:
‘’’
corrected_im = torch.nn.Parameter(corrected_im)
optimizer = optim.SGD([corrected_im],lr=1e-6,momentum=0.9)
optimizer.step()
‘’’
By the way , I still wonder what is changed by the nn.Parameter?