Hi:
Yeah, I used self.named_parameters() to iterate the name and parameters.
But how do I know, which name and parameters corresponding to the loglikelihood_grads calculated by
loglikelihood_grads = autograd.grad(loglikelihood, filter(lambda p: p.requires_grad, self.parameters()))