Update ener.py

iProzd · iProzd · commit a4d97a1203a8 · 2026-03-06T18:38:16.000+08:00
diff --git a/deepmd/pt/loss/ener.py b/deepmd/pt/loss/ener.py
@@ -556,6 +556,8 @@ def serialize(self) -> dict:
             "numb_generalized_coord": self.numb_generalized_coord,
             "use_huber": self.use_huber,
             "huber_delta": self.huber_delta,
+            "use_mae_loss": self.use_mae_loss,
+            "f_use_norm": self.f_use_norm,
         }
 
     @classmethod

Original file line number	Diff line number	Diff line change
`@@ -556,6 +556,8 @@ def serialize(self) -> dict:`
`556`	`556`	`"numb_generalized_coord": self.numb_generalized_coord,`
`557`	`557`	`"use_huber": self.use_huber,`
`558`	`558`	`"huber_delta": self.huber_delta,`
	`559`	`+ "use_mae_loss": self.use_mae_loss,`
	`560`	`+ "f_use_norm": self.f_use_norm,`
`559`	`561`	`}`
`560`	`562`
`561`	`563`	`@classmethod`