tf bacend for synthesis

2026-09-12 12:11:47 +08:00 · 2020-05-12 16:23:32 +02:00
parent d99fda8e42
commit b3ec50b5c4
2 changed files with 70 additions and 22 deletions
@@ -61,7 +61,6 @@ def visualize(alignment, postnet_output, stop_tokens, text, hop_length, CONFIG,
    plt.yticks(range(len(text)), list(text))
    plt.colorbar()
    # plot stopnet predictions
-    stop_tokens = stop_tokens.squeeze().detach().to('cpu').numpy()
    plt.subplot(num_plot, 1, 2)
    plt.plot(range(len(stop_tokens)), list(stop_tokens))
    # plot postnet spectrogram