LilianYe opened this issue 6 years ago · 0 comments
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])