Updates

yebiny · Oct 30, 2020 · 5ed6ee4 · 5ed6ee4
1 parent c45e2a6
commit 5ed6ee4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dataset/libri/preprocess.py b/dataset/libri/preprocess.py
@@ -53,6 +53,6 @@ def generate_transcript_file(dataset_name, transcripts):
         for transcript in transcripts:
             audio, transcript = transcript.split('|')
             text = " ".join(sp.EncodeAsPieces(transcript))
-            label = " ".join([str(item) for item in text])
+            label = " ".join([str(item) for item in sp.EncodeAsIds(transcript)])
 
             f.write('%s\t%s\t%s\n' % (audio, text, label))