make student_model_path optional in distillation (#173)

Signed-off-by: zjgemi <liuxin_zijian@163.com> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
deepmodeling · Oct 10, 2023 · 2bcc460 · 2bcc460
1 parent 8e8bc16
commit 2bcc460
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/dpgen2/entrypoint/args.py b/dpgen2/entrypoint/args.py
@@ -48,9 +48,7 @@ def dp_dist_train_args():
         Argument(
             "template_script", [list, str], optional=False, doc=doc_template_script
         ),
-        Argument(
-            "student_model_path", str, optional=False, doc=dock_student_model_path
-        ),
+        Argument("student_model_path", str, optional=True, doc=dock_student_model_path),
     ]
 
 

diff --git a/dpgen2/entrypoint/submit.py b/dpgen2/entrypoint/submit.py
@@ -436,7 +436,11 @@ def workflow_concurrent_learning(
                 "not match numb_models={numb_models}"
             )
     elif train_style == "dp-dist" and not old_style:
-        init_models_paths = [config["train"].get("student_model_path", None)]
+        init_models_paths = (
+            [config["train"]["student_model_path"]]
+            if "student_model_path" in config["train"]
+            else None
+        )
         config["train"]["numb_models"] = 1
     else:
         raise RuntimeError(