Commit 6ba19a97 authored by Yuxin Wu's avatar Yuxin Wu

fix ctc_loss API (#215)

parent 460addf8
......@@ -56,7 +56,7 @@ class Model(ModelDesc):
W_init=tf.truncated_normal_initializer(stddev=0.01))
logits = tf.reshape(logits, (BATCH, -1, NR_CLASS))
loss = tf.nn.ctc_loss(logits, label, seqlen, time_major=False)
loss = tf.nn.ctc_loss(label, logits, seqlen, time_major=False)
self.cost = tf.reduce_mean(loss, name='cost')
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment