add stat_file to argcheck

Signed-off-by: Jinzhe Zeng <[email protected]>
deepmodeling · Feb 28, 2024 · cfc9fe3 · cfc9fe3
1 parent 8caa6d6
commit cfc9fe3
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/deepmd/utils/argcheck.py b/deepmd/utils/argcheck.py
@@ -2117,6 +2117,7 @@ def training_args():  # ! modified by Ziyao: data configuration isolated.
         Argument("fitting_weight", dict, optional=True, doc=doc_fitting_weight),
         Argument("warmup_steps", int, optional=True, doc=doc_only_pt_supported),
         Argument("gradient_max_norm", float, optional=True, doc=doc_only_pt_supported),
+        Argument("stat_file", str, optional=True, doc=doc_only_pt_supported),
     ]
     variants = [
         Variant(