A quick follow up question - since Module doesn’t have an explicit  .backward() method, how do I exactly backprop on a loss function that is a module? Is it enough if I just use .train() instead?
Thanks!