From eb54e693259272c72bbc8bafbf94e9dbdd769692 Mon Sep 17 00:00:00 2001 From: Your Name Date: Sun, 10 Nov 2024 17:50:37 +0800 Subject: [PATCH] fix(tf): fix normalize when compressing a model converted from other backends (cherry picked from commit 232f7cdc4f3cdabe4ba8679b00ebb6d3ab3e75be) --- deepmd/tf/entrypoints/compress.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/deepmd/tf/entrypoints/compress.py b/deepmd/tf/entrypoints/compress.py index 6f3ff13e3b..24982d3120 100644 --- a/deepmd/tf/entrypoints/compress.py +++ b/deepmd/tf/entrypoints/compress.py @@ -147,6 +147,8 @@ def compress( 10 * step, int(frequency), ] + jdata.setdefault("training", {"numb_steps": 0}) + jdata.setdefault("learning_rate", {}) jdata["training"]["save_ckpt"] = os.path.join("model-compression", "model.ckpt") jdata = update_deepmd_input(jdata) jdata = normalize(jdata)