I would like to know what are the best practices/tools to visualize what happens during loss.backward().
I am confident in my understanding of the forward pass of my model, how can I control its backward pass?