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

bug fix in testing phase

This commit is contained in:
Eren Golge 2018-11-21 15:42:10 +01:00
parent 3a99ee9492
commit 36efaa789d
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
train.py
View File

@ -330,7 +330,7 @@ def evaluate(model, criterion, criterion_st, data_loader, ap, current_step):
ap.griffin_lim_iters = 60 ap.griffin_lim_iters = 60
for idx, test_sentence in enumerate(test_sentences): for idx, test_sentence in enumerate(test_sentences):
try: try:
wav, alignment, linear_spec, stop_tokens = synthesis( wav, alignment, linear_spec, _, stop_tokens = synthesis(
model, test_sentence, c, use_cuda, ap) model, test_sentence, c, use_cuda, ap)
file_path = os.path.join(AUDIO_PATH, str(current_step)) file_path = os.path.join(AUDIO_PATH, str(current_step))
@ -345,6 +345,7 @@ def evaluate(model, criterion, criterion_st, data_loader, ap, current_step):
wav, wav,
current_step, current_step,
sample_rate=c.audio['sample_rate']) sample_rate=c.audio['sample_rate'])
linear_spec = plot_spectrogram(linear_spec, ap) linear_spec = plot_spectrogram(linear_spec, ap)
align_img = plot_alignment(alignment) align_img = plot_alignment(alignment)
tb.add_figure('TestSentences/{}_Spectrogram'.format(idx), tb.add_figure('TestSentences/{}_Spectrogram'.format(idx),
@ -352,7 +353,8 @@ def evaluate(model, criterion, criterion_st, data_loader, ap, current_step):
tb.add_figure('TestSentences/{}_Alignment'.format(idx), align_img, tb.add_figure('TestSentences/{}_Alignment'.format(idx), align_img,
current_step) current_step)
except: except:
print(" !! Error as creating Test Sentence -", idx) print(" !! Error creating Test Sentence -", idx)
traceback.print_exc()
pass pass
return avg_linear_loss return avg_linear_loss