use new implementation

wanghan-iapcm · Apr 9, 2024 · 9a0b366 · 9a0b366
1 parent 1285105
commit 9a0b366
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dpgen2/op/run_dp_train.py b/dpgen2/op/run_dp_train.py
@@ -322,7 +322,7 @@ def clean_before_quit():
                 json.dump(train_dict, fp, indent=4)
 
             # train model
-            command = _make_train_command_old(
+            command = _make_train_command(
                 dp_command,
                 train_script_name,
                 impl,