Closed LilianYe closed 6 years ago
In function xentropy_loss() of model.py logits = tf.reshape(logits, [tf.shape(logits)[0], -1, 2]) should be logits = tf.reshape(logits, [tf.shape(logits)[0], -1, self.num_classes])
logits = tf.reshape(logits, [tf.shape(logits)[0], -1, 2])
logits = tf.reshape(logits, [tf.shape(logits)[0], -1, self.num_classes])
In function xentropy_loss() of model.py
logits = tf.reshape(logits, [tf.shape(logits)[0], -1, 2])should belogits = tf.reshape(logits, [tf.shape(logits)[0], -1, self.num_classes])