Update dpgen/generator/run.py

Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com> Signed-off-by: Jinzhe Zeng <[email protected]>
deepmodeling · Nov 15, 2024 · e2af938 · e2af938
1 parent 10b9399
commit e2af938
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dpgen/generator/run.py b/dpgen/generator/run.py
@@ -767,7 +767,7 @@ def run_train_dp(iter_index, jdata, mdata):
     if suffix == ".pth":
         train_command += " --pt"
     elif suffix == ".savedmodel":
-        train_command += "--jax"
+        train_command += " --jax"
 
     # paths
     iter_name = make_iter_name(iter_index)