xnn.common.train.losses

Weighted energy / force / stress loss.

Functions

weighted_loss(pred, data, energy_weight, ...)

Compute a weighted sum of energy, force and stress loss terms.