fix the text_data dim to fit the dim check in paddle develop version

5 years ago · 5522621b46
parent 517c027c5e
commit 5522621b46
1 changed files with 2 additions and 3 deletions
--- a/data_utils/data.py
+++ b/data_utils/data.py
@ -330,11 +330,10 @@ class DataGenerator(object):
                axis=0)
            masks.append(mask)
        padded_audios = np.array(padded_audios).astype('float32')
+        texts = np.expand_dims(np.array(texts).astype('int32'), axis=-1)
        if self._is_training:
            texts = fluid.create_lod_tensor(
-                np.array(texts).astype('int32'),
-                recursive_seq_lens=[text_lens],
-                place=self._place)
+                texts, recursive_seq_lens=[text_lens], place=self._place)
        audio_lens = np.array(audio_lens).astype('int64').reshape([-1, 1])
        masks = np.array(masks).astype('float32')
        return padded_audios, texts, audio_lens, masks