From a5c81c75d5ed69607447b675ef9d004504a935b2 Mon Sep 17 00:00:00 2001 From: lfchener Date: Tue, 15 Dec 2020 11:27:11 +0000 Subject: [PATCH] fix add_spectrogram_plots --- parakeet/utils/display.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/parakeet/utils/display.py b/parakeet/utils/display.py index b552a94..6c13931 100644 --- a/parakeet/utils/display.py +++ b/parakeet/utils/display.py @@ -93,7 +93,7 @@ def add_multi_attention_plots(writer, tag, attention_weights, global_step): def add_spectrogram_plots(writer, tag, spec, global_step): - spec = spec.numpy() + spec = spec.numpy().T fig, ax = plt.subplots(figsize=(12, 3)) im = ax.imshow(spec, aspect="auto", origin="lower", interpolation='none') plt.colorbar(im, ax=ax)