bug fix for plotting

pull/10/head
Eren 2018-08-11 16:09:47 +02:00
parent 3b2654203d
commit b65f5bd1d0
2 changed files with 3 additions and 3 deletions

View File

@ -15,7 +15,7 @@
"text_cleaner": "english_cleaners",
"epochs": 1000,
"lr": 0.002,
"lr": 0.001,
"lr_decay": 0.5,
"decay_step": 100000,
"warmup_steps": 4000,

View File

@ -312,9 +312,9 @@ def evaluate(model, criterion, criterion_st, data_loader, ap, current_step):
# test sentences
ap.griffin_lim_iters = 60
for idx, test_sentence in enumerate(test_sentences):
try:
wav, linear_spec, alignments = synthesis(model, ap, test_sentence,
wav, linear_spec, alignments = synthesis(model, ap, test_sentence,
use_cuda, c.text_cleaner)
try:
wav_name = 'TestSentences/{}'.format(idx)
tb.add_audio(
wav_name, wav, current_step, sample_rate=c.sample_rate)