Dang it does! Thanks! You also made me realise that I was using the wrong target anyways. Like its literally in the comment above to use D_T_G_z  But I gotta ask…are the weights also backprogagated for the target in a loss function? Because the issue seems to be that the gradients were backprogated twice for the same evaluation for D_T_x. If you could help me clear this up, I would deeply appreciate it.