coqui-ai
/
TTS
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Rename utils due to name conflict

pull/10/head
Eren 2018-07-05 17:02:14 +02:00
parent bdf3fd209a
commit 83707b4b93
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
notebooks/utils.py → notebooks/synthesis.py
View File

@ -14,7 +14,7 @@ def create_speech(m, s, CONFIG, use_cuda, ap):
chars_var = torch.from_numpy(seq).unsqueeze(0)
if use_cuda:
chars_var = chars_var.cuda()
mel_out, linear_out, alignments, stop_tokens = m.forward(chars_var)
mel_out, linear_out, alignments, stop_tokens = m.forward(chars_var.long())
linear_out = linear_out[0].data.cpu().numpy()
alignment = alignments[0].cpu().data.numpy()
spec = ap._denormalize(linear_out)
@ -34,7 +34,7 @@ def visualize(alignment, spectrogram, stop_tokens, CONFIG):
plt.xlabel("Decoder timestamp", fontsize=label_fontsize)
plt.ylabel("Encoder timestamp", fontsize=label_fontsize)
plt.colorbar()
stop_tokens = stop_tokens.squeeze().detach().to('cpu').numpy()
plt.subplot(3, 1, 2)
plt.plot(range(len(stop_tokens)), list(stop_tokens))
@ -46,5 +46,3 @@ def visualize(alignment, spectrogram, stop_tokens, CONFIG):
plt.ylabel("Hz", fontsize=label_fontsize)
plt.tight_layout()
plt.colorbar()