testadmin
/
Parakeet
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix add_spectrogram_plots

This commit is contained in:
lfchener 2020-12-15 11:27:11 +00:00
parent c864612dc3
commit a5c81c75d5
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
parakeet/utils/display.py
View File

@ -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)