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,
|
train_avg_loss_dict, global_step = train(model, criterion, optimizer,
|
||||||
optimizer_st, scheduler, ap,
|
optimizer_st, scheduler, ap,
|
||||||
global_step, epoch, amp, speaker_mapping)
|
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)
|
c_logger.print_epoch_end(epoch, eval_avg_loss_dict)
|
||||||
target_loss = train_avg_loss_dict['avg_postnet_loss']
|
target_loss = train_avg_loss_dict['avg_postnet_loss']
|
||||||
if c.run_eval:
|
if c.run_eval:
|
||||||
|
|
Loading…
Reference in New Issue