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

remove debug code

This commit is contained in:
erogol 2020-08-14 16:14:11 +02:00
parent 77c577ddab
commit 09ad6a09b0
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
mozilla_voice_tts/bin/train_glow_tts.py
View File

@ -547,10 +547,9 @@ def main(args): # pylint: disable=redefined-outer-name
model = data_depended_init(model, ap)
for epoch in range(0, c.epochs):
c_logger.print_epoch_start(epoch, c.epochs)
# train_avg_loss_dict, global_step = train(model, criterion, optimizer,
# scheduler, ap, global_step,
# epoch, amp)
train_avg_loss_dict, global_step = 0, 0
train_avg_loss_dict, global_step = train(model, criterion, optimizer,
scheduler, ap, global_step,
epoch, amp)
eval_avg_loss_dict = evaluate(model, criterion, ap, global_step, epoch)
c_logger.print_epoch_end(epoch, eval_avg_loss_dict)
target_loss = train_avg_loss_dict['avg_loss']