"""Training loss modules."""