mirror of https://github.com/coqui-ai/TTS.git
Merge pull request #498 from thllwg/dev
Missing speaker_mapping argument causes training to break
This commit is contained in:
commit
81dd3629cb
|
@ -624,7 +624,7 @@ def main(args): # pylint: disable=redefined-outer-name
|
|||
train_avg_loss_dict, global_step = train(model, criterion, optimizer,
|
||||
optimizer_st, scheduler, ap,
|
||||
global_step, epoch, amp, speaker_mapping)
|
||||
eval_avg_loss_dict = evaluate(model, criterion, ap, global_step, epoch)
|
||||
eval_avg_loss_dict = evaluate(model, criterion, ap, global_step, epoch, speaker_mapping)
|
||||
c_logger.print_epoch_end(epoch, eval_avg_loss_dict)
|
||||
target_loss = train_avg_loss_dict['avg_postnet_loss']
|
||||
if c.run_eval:
|
||||
|
|
Loading…
Reference in New Issue