add start_pref_pf and limit_pref_pf to loss Argument

fix #1199
deepmodeling · Oct 9, 2021 · 3d2da57 · 3d2da57
1 parent d8acbb8
commit 3d2da57
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/deepmd/utils/argcheck.py b/deepmd/utils/argcheck.py
@@ -478,6 +478,8 @@ def loss_ener():
     doc_limit_pref_v = limit_pref('virial')
     doc_start_pref_ae = start_pref('atom_ener')
     doc_limit_pref_ae = limit_pref('atom_ener')
+    doc_start_pref_pf = start_pref('atom_pref')
+    doc_limit_pref_pf = limit_pref('atom_pref')
     doc_relative_f = 'If provided, relative force error will be used in the loss. The difference of force will be normalized by the magnitude of the force in the label with a shift given by `relative_f`, i.e. DF_i / ( || F || + relative_f ) with DF denoting the difference between prediction and label and || F || denoting the L2 norm of the label.'
     return [
         Argument("start_pref_e", [float,int], optional = True, default = 0.02, doc = doc_start_pref_e),
@@ -488,6 +490,8 @@ def loss_ener():
         Argument("limit_pref_v", [float,int], optional = True, default = 0.00, doc = doc_limit_pref_v),
         Argument("start_pref_ae", [float,int], optional = True, default = 0.00, doc = doc_start_pref_ae),
         Argument("limit_pref_ae", [float,int], optional = True, default = 0.00, doc = doc_limit_pref_ae),
+        Argument("start_pref_pf", [float,int], optional = True, default = 0.00, doc = doc_start_pref_pf),
+        Argument("limit_pref_pf", [float,int], optional = True, default = 0.00, doc = doc_limit_pref_pf),
         Argument("relative_f", [float,None], optional = True, doc = doc_relative_f)
     ]