When would we choose to use the Module forms of the loss functions rather than the functional forms? For example, torch.nn.MSELoss over torch.nn.functional.mse_loss. I just realised that I’ve never used the Module forms, and I’m wondering whether I’m missing a useful trick or something? It seems to me that loss is something we would almost always want to calculate outside of the model, so it doesn’t make intuitive sense for it to be a Module.