I am currently confused as to where the zeroing of the gradients has to be performed.
is the zero_grad function called on the optimizer or on the module, i.e. the network. I have seen examples of both…