Hi,
I understood your problem, but it’s hard to tell with just three lines of code. You can debug your optimizer by checking opt.param_groups, it will give you all the parameter it tries to optimize. Hope this helps.