replace tf.no_op (#1419)

`tf.no_op` has a strange pefermance issue... We need to use another tensor as placeholder. (cherry picked from commit f3b7189)
deepmodeling · Jan 15, 2022 · d7795ca · d7795ca
1 parent e80e222
commit d7795ca
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepmd/loss/ener.py b/deepmd/loss/ener.py
@@ -129,7 +129,7 @@ def build (self,
         return l2_loss, more_loss
 
     def eval(self, sess, feed_dict, natoms):
-        placeholder = tf.no_op()
+        placeholder = self.l2_l
         run_data = [
             self.l2_l,
             self.l2_more['l2_ener_loss'] if self.has_e else placeholder,