Plot specs and alignments for test sentences

2026-09-09 11:16:00 +08:00 · 2018-07-23 16:15:26 +02:00
parent 766e88700d
commit c72f309068
5 changed files with 19 additions and 10 deletions
@@ -166,7 +166,7 @@ def synthesis(model, ap, text, use_cuda, text_cleaner):
        chars_var = torch.from_numpy(seq).unsqueeze(0)
        if use_cuda:
            chars_var = chars_var.cuda().long()
-        _, linear_out, _, _ = model.forward(chars_var)
+        _, linear_out, alignments, _ = model.forward(chars_var)
        linear_out = linear_out[0].data.cpu().numpy()
        wav = ap.inv_spectrogram(linear_out.T)
-        return wav
+        return wav, linear_out, alignments