fix compress

Signed-off-by: Jinzhe Zeng <[email protected]>
deepmodeling · May 20, 2024 · 426190f · 426190f
1 parent 50165bb
commit 426190f
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/deepmd/tf/train/trainer.py b/deepmd/tf/train/trainer.py
@@ -272,9 +272,7 @@ def _build_network(self, data, suffix=""):
                 self.place_holders[kk] = tf.placeholder(
                     GLOBAL_TF_FLOAT_PRECISION, [None], "t_" + kk
                 )
-            self._get_place_holders(
-                {rr.key: rr.dict for rr in self.model.input_requirement}
-            )
+            self._get_place_holders({rr.key: rr.dict for rr in self.data_requirements})
         else:
             self._get_place_holders(data.get_data_dict())