enerzyme.models.loss.RMSELoss#

class enerzyme.models.loss.RMSELoss(**weights: Dict[str, float])[source]#

Bases: WeightedLoss

__init__(**weights: Dict[str, float]) → None[source]#

loss_fn(output: Dict[str, Tensor], target: Dict[str, Tensor], k: str) → Tensor[source]#