Add stop token visualizaiotn for the notebook

2025-12-24 04:09:32 +01:00 · 2018-05-03 05:56:33 -07:00
parent f8d5bbd5d2
commit d02dd47958
2 changed files with 10 additions and 5 deletions
--- a/utils/visual.py
+++ b/utils/visual.py
@@ -32,4 +32,4 @@ def plot_spectrogram(linear_output, audio):
    data = np.fromstring(fig.canvas.tostring_rgb(), dtype=np.uint8, sep='')
    data = data.reshape(fig.canvas.get_width_height()[::-1] + (3,))
    plt.close()
-    return data
+    return data