diff --git a/config.json b/config.json index 2b4c1441..148ab3db 100644 --- a/config.json +++ b/config.json @@ -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, diff --git a/train.py b/train.py index bfe00aee..b9d1616e 100644 --- a/train.py +++ b/train.py @@ -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)