I\'m trying to make a custom training loop for a network that has multiple inputs and outputs, I found that the cause of my issues is from having multiple outputs. I dont u
