修正了参数名不对应的问题

Hcracker-luke · Jun 18, 2017 · b35d210 · b35d210
1 parent 978457c
commit b35d210
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tensorflowTUT/tf20_RNN2.2/full_code.py b/tensorflowTUT/tf20_RNN2.2/full_code.py
@@ -100,8 +100,8 @@ def compute_cost(self):
                 name='average_cost')
             tf.summary.scalar('cost', self.cost)
 
-    def ms_error(self, y_target, y_pre):
-        return tf.square(tf.subtract(y_target, y_pre))
+    def ms_error(self, labels, logits):
+        return tf.square(tf.subtract(labels, logits))
 
     def _weight_variable(self, shape, name='weights'):
         initializer = tf.random_normal_initializer(mean=0., stddev=1.,)