Feat: output logs on freezing models (#4165)

`dp --pt freeze` does not output whether the operation finishes successfully. This PR adds a log message on it.
deepmodeling · Sep 27, 2024 · ab48d90 · ab48d90
1 parent 0b3f860
commit ab48d90
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/deepmd/pt/entrypoints/main.py b/deepmd/pt/entrypoints/main.py
@@ -354,6 +354,7 @@ def freeze(FLAGS):
         FLAGS.output,
         extra_files,
     )
+    log.info(f"Saved frozen model to {FLAGS.output}")
 
 
 def change_bias(FLAGS):