Thanks for your reply. Yes of course I’ve set it…I think that probably there’s something under the hood made by PL which lost the computational graph linked to the model parameters, hence the .grad return all zeros.
Thanks for your reply. Yes of course I’ve set it…I think that probably there’s something under the hood made by PL which lost the computational graph linked to the model parameters, hence the .grad return all zeros.