diff --git a/deepmd/train/trainer.py b/deepmd/train/trainer.py
index 40583e6223..77d5028051 100644
--- a/deepmd/train/trainer.py
+++ b/deepmd/train/trainer.py
@@ -699,4 +699,6 @@ def _init_from_frz_model(self):
             ) from e
         else:
             self.model_type = bytes.decode(t_model_type)
+        if self.model_type == 'compressed_model':
+            self.frz_model = self.run_opt.init_frz_model
         self.model.init_variables(graph, graph_def, model_type=self.model_type)