So when I clock all of the lines (https://pastebin.com/Q1nXMStH 22), it shows that the slowdown happens in the loss.backward()-function. I guess one way for that function would slow down is if the graph “grows” with each iteration. How would I see the computation graph that is calculated there? Any idea how to fix that code? Thanks!