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

plot float16 alignments

This commit is contained in:
erogol 2020-12-28 13:53:51 +01:00
parent 13c6665c92
commit e4680e1b99
1 changed files with 2 additions and 0 deletions
TTS/tts/utils
visual.py

2
TTS/tts/utils/visual.py
View File

@ -17,6 +17,8 @@ def plot_alignment(alignment,
alignment_ = alignment.detach().cpu().numpy().squeeze() alignment_ = alignment.detach().cpu().numpy().squeeze()
else: else:
alignment_ = alignment alignment_ = alignment
alignment_ = alignment_.astype(
np.float32) if alignment_.dtype == np.float16 else alignment_
fig, ax = plt.subplots(figsize=fig_size) fig, ax = plt.subplots(figsize=fig_size)
im = ax.imshow(alignment_.T, im = ax.imshow(alignment_.T,
aspect='auto', aspect='auto',