From fadea4f3779cc4ded37b4c3f37b03e57fc72eb12 Mon Sep 17 00:00:00 2001 From: Jinzhe Zeng Date: Tue, 3 May 2022 16:28:52 -0400 Subject: [PATCH] fix compress training Fix #1679. The bug was introduced in #1610. --- deepmd/train/trainer.py | 2 ++ 1 file changed, 2 insertions(+) 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)