Convert embeds to np array (#895)

AryanVBW · Nov 21, 2021 · b7a6674 · b7a6674
1 parent 7432046
commit b7a6674
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/synthesizer/synthesizer_dataset.py b/synthesizer/synthesizer_dataset.py
@@ -72,7 +72,7 @@ def collate_synthesizer(batch, r, hparams):
     mel = np.stack(mel)
 
     # Speaker embedding (SV2TTS)
-    embeds = [x[2] for x in batch]
+    embeds = np.array([x[2] for x in batch])
 
     # Index (for vocoder preprocessing)
     indices = [x[3] for x in batch]