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

final fixes

This commit is contained in:
gerazov 2021-02-06 23:05:01 +01:00 committed by Eren Gölge
parent 10d5a63d49
commit b3c5cc2cdc
2 changed files with 5 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
TTS/bin/train_glow_tts.py
View File

@ -552,7 +552,7 @@ def main(args): # pylint: disable=redefined-outer-name
train_avg_loss_dict, global_step = train(train_loader, model, criterion, optimizer, train_avg_loss_dict, global_step = train(train_loader, model, criterion, optimizer,
scheduler, ap, global_step, scheduler, ap, global_step,
epoch) epoch)
eval_avg_loss_dict = evaluate(eval_loader , model, criterion, ap, eval_avg_loss_dict = evaluate(eval_loader, model, criterion, ap,
global_step, epoch) global_step, epoch)
c_logger.print_epoch_end(epoch, eval_avg_loss_dict) c_logger.print_epoch_end(epoch, eval_avg_loss_dict)
target_loss = train_avg_loss_dict['avg_loss'] target_loss = train_avg_loss_dict['avg_loss']

10
TTS/bin/train_tacotron.py
View File

@ -56,12 +56,10 @@ def setup_loader(ap, r, is_val=False, verbose=False, dataset=None):
phoneme_language=c.phoneme_language, phoneme_language=c.phoneme_language,
enable_eos_bos=c.enable_eos_bos_chars, enable_eos_bos=c.enable_eos_bos_chars,
verbose=verbose, verbose=verbose,
speaker_mapping=( speaker_mapping=(speaker_mapping if (
speaker_mapping if ( c.use_speaker_embedding
c.use_speaker_embedding and and c.use_external_speaker_embedding_file
c.use_external_speaker_embedding_file ) else None)
) else None
)
) )
if c.use_phonemes and c.compute_input_seq_cache: if c.use_phonemes and c.compute_input_seq_cache: