Pytorch Multiple Loss Functions