Skip to content

Commit 087179a

Browse files
committed
✍️ initialize data pipeline in pure tf
1 parent af34109 commit 087179a

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

tensorflow_asr/datasets/keras/asr_dataset.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -200,7 +200,7 @@ def create(self, batch_size):
200200
return self.process(dataset, batch_size)
201201

202202

203-
class TFASRTFRecordDatasetKeras(ASRDatasetKeras):
203+
class TFASRTFRecordDatasetKeras(ASRTFRecordDatasetKeras):
204204
def preprocess(self, audio, indices):
205205
with tf.device("/CPU:0"):
206206
signal = tf_read_raw_audio(audio, self.speech_featurizer.sample_rate)
@@ -279,3 +279,6 @@ def create(self, batch_size):
279279
entries = np.delete(entries, 1, 1) # Remove unused duration
280280
dataset = tf.data.Dataset.from_tensor_slices(entries)
281281
return self.process(dataset, batch_size)
282+
283+
284+
class TFASRSliceDatasetKeras(ASRSliceDatasetKeras):

0 commit comments

Comments
 (0)