Hi Diego,
Of course, I forgot this line. However, still any gradients aren’t computed. As I mentioned in the first post, the problem is with None values of parameters’ grads after  calling .backward function, not actually with applying the updates with correctly computed gradients (which is done by optimizer.step()).