Modularize functions in Tacotron

This commit is contained in:
Eren Golge
2019-03-05 13:25:50 +01:00
parent cc34fe4c7c
commit 1e8fdec084
5 changed files with 121 additions and 91 deletions
+1 -1
View File
@@ -20,7 +20,7 @@ def synthesis(m, s, CONFIG, use_cuda, ap):
chars_var = torch.from_numpy(seq).unsqueeze(0)
if use_cuda:
chars_var = chars_var.cuda()
mel_spec, linear_spec, alignments, stop_tokens = m.forward(
mel_spec, linear_spec, alignments, stop_tokens = m.inference(
chars_var.long())
linear_spec = linear_spec[0].data.cpu().numpy()
mel_spec = mel_spec[0].data.cpu().numpy()