Skip to content

S
seminar-breakout
Commit 6ba19a97 authored by Yuxin Wu's avatar Yuxin Wu
Browse files
Options

fix ctc_loss API (#215)

parent 460addf8
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/CTC-TIMIT/train-timit.py
View file @ 6ba19a97
...@@ -56,7 +56,7 @@ class Model(ModelDesc): ...@@ -56,7 +56,7 @@ class Model(ModelDesc):
W_init=tf.truncated_normal_initializer(stddev=0.01)) W_init=tf.truncated_normal_initializer(stddev=0.01))
logits = tf.reshape(logits, (BATCH, -1, NR_CLASS)) 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') 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