fix bug of loss_func

deepmodeling · Mar 22, 2024 · 1818271 · 1818271
1 parent 09d4de1
commit 1818271
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/deepmd/pt/loss/property.py b/deepmd/pt/loss/property.py
@@ -93,19 +93,19 @@ def forward(self, input_dict, model, label, natoms, learning_rate, mae=False):
                 reduction="sum",
                 beta=self.beta,
             )
-        elif self.func == "mae":
+        elif self.loss_func == "mae":
             loss += F.l1_loss(
                 (label["property"] - label_mean) / label_std,
                 model_pred["property"],
                 reduction="sum",
             )
-        elif self.func == "mse":
+        elif self.loss_func == "mse":
             loss += F.mse_loss(
                 (label["property"] - label_mean) / label_std,
                 model_pred["property"],
                 reduction="sum",
             )
-        elif self.func == "rmse":
+        elif self.loss_func == "rmse":
             loss += torch.sqrt(
                 F.mse_loss(
                     (label["property"] - label_mean) / label_std,