From 76917831c050882055b009c01a866b8a98445ad8 Mon Sep 17 00:00:00 2001 From: Jinzhe Zeng Date: Sat, 28 Aug 2021 04:40:58 -0400 Subject: [PATCH] fix hybrid descriptor training error fix #1051. --- deepmd/train/trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deepmd/train/trainer.py b/deepmd/train/trainer.py index e301a0753b..72dc61418e 100644 --- a/deepmd/train/trainer.py +++ b/deepmd/train/trainer.py @@ -315,7 +315,7 @@ def build (self, self._init_from_frz_model() self.neighbor_stat \ - = NeighborStat(self.ntypes, self.descrpt_param['rcut']) + = NeighborStat(self.ntypes, self.descrpt.get_rcut()) self.min_nbor_dist, self.max_nbor_size \ = self.neighbor_stat.get_stat(data) tf.constant(self.min_nbor_dist,