From ab48d90bb9076c7e2278f8414129820342d3c31b Mon Sep 17 00:00:00 2001
From: Chun Cai <amoycaic@gmail.com>
Date: Fri, 27 Sep 2024 08:38:11 +0800
Subject: [PATCH] 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.
---
 deepmd/pt/entrypoints/main.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/deepmd/pt/entrypoints/main.py b/deepmd/pt/entrypoints/main.py
index 9133575ec..3df05cbb4 100644
--- 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):