Skip to content

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

fix train-timit (#603)

parent e4953ecf
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 3 deletions
examples/CTC-TIMIT/train-timit.py
View file @ bc8da700
...@@ -10,6 +10,7 @@ from six.moves import range ...@@ -10,6 +10,7 @@ from six.moves import range
from tensorpack import * from tensorpack import *
from tensorpack.tfutils.gradproc import SummaryGradient, GlobalNormClip from tensorpack.tfutils.gradproc import SummaryGradient, GlobalNormClip
from tensorpack.utils import serialize
import tensorflow as tf import tensorflow as tf
from timitdata import TIMITBatch from timitdata import TIMITBatch
...@@ -26,9 +27,9 @@ FEATUREDIM = 39 # MFCC feature dimension ...@@ -26,9 +27,9 @@ FEATUREDIM = 39 # MFCC feature dimension
class Model(ModelDesc): class Model(ModelDesc):
def _get_inputs(self): def _get_inputs(self):
return [InputDesc(tf.float32, [None, None, FEATUREDIM], 'feat'), # bxmaxseqx39 return [InputDesc(tf.float32, [None, None, FEATUREDIM], 'feat'), # bxmaxseqx39
InputDesc(tf.int64, None, 'labelidx'), # label is b x maxlen, sparse InputDesc(tf.int64, [None, None], 'labelidx'), # label is b x maxlen, sparse
InputDesc(tf.int32, None, 'labelvalue'), InputDesc(tf.int32, [None], 'labelvalue'),
InputDesc(tf.int64, None, 'labelshape'), InputDesc(tf.int64, [None], 'labelshape'),
InputDesc(tf.int32, [None], 'seqlen'), # b InputDesc(tf.int32, [None], 'seqlen'), # b
] ]
......
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