From e2af9388c5e367efef3667ed151e1f65cfe5f53f Mon Sep 17 00:00:00 2001 From: Jinzhe Zeng Date: Fri, 15 Nov 2024 15:22:41 -0500 Subject: [PATCH] Update dpgen/generator/run.py Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com> Signed-off-by: Jinzhe Zeng --- dpgen/generator/run.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dpgen/generator/run.py b/dpgen/generator/run.py index 9d827e98a..b7c4033a2 100644 --- 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)