From 3d2da57f123cffb1132233c29e3305d697101b7d Mon Sep 17 00:00:00 2001 From: Jinzhe Zeng Date: Fri, 8 Oct 2021 22:38:45 -0400 Subject: [PATCH] add `start_pref_pf` and `limit_pref_pf` to `loss` Argument fix #1199 --- deepmd/utils/argcheck.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/deepmd/utils/argcheck.py b/deepmd/utils/argcheck.py index 0ea84d8e73..857dc04e10 100644 --- 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) ]