acoustic: Fix return shapes

revsic · Feb 9, 2022 · 465fb53 · 465fb53
1 parent a2e3e65
commit 465fb53
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/speeches/acoustic.py b/speeches/acoustic.py
@@ -69,4 +69,4 @@ def collate(self, bunch: List[Tuple[np.ndarray, np.ndarray]]) \
         # [B, T, mel]
         mel = np.stack(
             [np.pad(spec, [[0, mellen.max() - len(spec)], [0, 0]]) for _, spec in bunch])
-        return mel, text, textlen, mellen
+        return text, mel, textlen, mellen