Update train_tts.py

align style_input with the new config key "gst_style_input"
This commit is contained in:
Alexander Korolev 2020-08-14 09:38:53 +02:00 committed by GitHub
parent 96815bfdf3
commit 2a2a0864c1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions

View File

@ -441,7 +441,7 @@ def evaluate(model, criterion, ap, global_step, epoch, speaker_mapping=None):
test_figures = {}
print(" | > Synthesizing test sentences")
speaker_id = 0 if c.use_speaker_embedding else None
style_wav = c.get("style_wav_for_test")
style_wav = c.get("gst_style_input")
for idx, test_sentence in enumerate(test_sentences):
try:
wav, alignment, decoder_output, postnet_output, stop_tokens, _ = synthesis(