also if I do: loss = Variable(loss, requires_grad = True) it does run but
you’re right in that the loss over 50 epochs just fluctuates and stays the same on the training set, there’s no convergence of loss, so the gradients seem to be getting messed up.