revert config loss stress

ACEsuit · Aug 23, 2024 · 5e70bfe · 5e70bfe
1 parent a0b2af3
commit 5e70bfe
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mace/modules/loss.py b/mace/modules/loss.py
@@ -282,8 +282,8 @@ def forward(self, ref: Batch, pred: TensorDict) -> torch.Tensor:
             * conditional_huber_forces(ref, pred, huber_delta=self.huber_delta)
             + self.stress_weight
             * self.huber_loss(
-                configs_weight * configs_stress_weight * ref["stress"],
-                configs_weight * configs_stress_weight * pred["stress"],
+                ref["stress"],
+                pred["stress"],
             )
         )