Hi, Thanks for your reply.
I understand that in Pytorch I just need to pass the parameters. However here the loss is not scalar! When I ask to back prob ’ self.select_loss’ , Pytorch says it needs to be scalar!
I want to find gradient of the self.select_loss for against all the model variables