Feat: output logs on freezing models (deepmodeling#4165)

caic99 · web-flow · commit ab48d90bb907 · 2024-09-27T00:38:11.000Z
`dp --pt freeze` does not output whether the operation finishes
successfully. This PR adds a log message on it.
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):

Original file line number	Diff line number	Diff line change
`@@ -354,6 +354,7 @@ def freeze(FLAGS):`
`354`	`354`	`FLAGS.output,`
`355`	`355`	`extra_files,`
`356`	`356`	`)`
	`357`	`+ log.info(f"Saved frozen model to {FLAGS.output}")`
`357`	`358`
`358`	`359`
`359`	`360`	`def change_bias(FLAGS):`