make numb_models as common argument

Signed-off-by: Jinzhe Zeng <jinzhe.zeng@rutgers.edu>

Author: njzjz

dpgen/generator/arginfo.py

@@ -78,6 +78,12 @@ def data_args() -> list[Argument]:
 
 # Training
 
+def training_args_common() -> list[Argument]:
+    doc_numb_models = "Number of models to be trained in 00.train. 4 is recommend."
+    return [
+        Argument("numb_models", int, optional=False, doc=doc_numb_models),
+    ]
+
 
 def training_args_dp() -> list[Argument]:
     """Traning arguments.
@@ -90,7 +96,6 @@ def training_args_dp() -> list[Argument]:
     doc_train_backend = (
         "The backend of the training. Currently only support tensorflow and pytorch."
     )
-    doc_numb_models = "Number of models to be trained in 00.train. 4 is recommend."
     doc_training_iter0_model_path = "The model used to init the first iter training. Number of element should be equal to numb_models."
     doc_training_init_model = "Iteration > 0, the model parameters will be initilized from the model trained at the previous iteration. Iteration == 0, the model parameters will be initialized from training_iter0_model_path."
     doc_default_training_param = "Training parameters for deepmd-kit in 00.train. You can find instructions from `DeePMD-kit documentation <https://docs.deepmodeling.org/projects/deepmd/>`_."
@@ -133,7 +138,6 @@ def training_args_dp() -> list[Argument]:
             default="tensorflow",
             doc=doc_train_backend,
         ),
-        Argument("numb_models", int, optional=False, doc=doc_numb_models),
         Argument(
             "training_iter0_model_path",
             list[str],
@@ -999,7 +1003,7 @@ def run_jdata_arginfo() -> Argument:
     return Argument(
         "run_jdata",
         dict,
-        sub_fields=basic_args() + data_args() + fp_args(),
+        sub_fields=basic_args() + data_args() + training_args_common() + fp_args(),
         sub_variants=[training_args(), *model_devi_args(), fp_style_variant_type_args()],
         doc=doc_run_jdata,
     )

dpgen/simplify/arginfo.py

@@ -12,6 +12,7 @@
     fp_style_siesta_args,
     fp_style_vasp_args,
     training_args,
+    training_args_common,
 )
 
 
@@ -201,6 +202,7 @@ def simplify_jdata_arginfo() -> Argument:
             *data_args(),
             *general_simplify_arginfo(),
             # simplify use the same training method as run
+            *training_args_common(),
             *fp_args(),
         ],
         sub_variants=[