I’m using PyTorch to solve an optimization problem with Lagrangian relaxation. I think my implementation is correct but it just does not converge and I do not why. I’m unsure what the problem is and what I lost or forgot.