fix hybrid descriptor training error

fix deepmodeling#1051.
njzjz · Aug 28, 2021 · 7691783 · 7691783
1 parent 37af1d1
commit 7691783
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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,