https://machinelearningmastery.com/loss-functions-in-pytorch-models/