model.pt -> model.ckpt.pt

Signed-off-by: zjgemi <[email protected]>
wangzyphysics · Mar 22, 2024 · a409d47 · a409d47
1 parent 10ba65a
commit a409d47
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
@@ -272,7 +272,7 @@ def clean_before_quit():
                     raise FatalError("dp freeze failed")
                 model_file = "frozen_model.pb"
             elif impl == "pytorch":
-                model_file = "model.pt"
+                model_file = "model.ckpt.pt"
             fplog.write("#=================== freeze std out ===================\n")
             fplog.write(out)
             fplog.write("#=================== freeze std err ===================\n")