I tried every possible way but the loss remains constant. Checked the computation graph, loss functions dimensions… I feel like I’m struck with this. I’m trying to replicate this https://www.hindawi.com/journals/complexity/2020/6367273/ 2