wassname/TTS
1
0
Fork 0
mirror of https://github.com/wassname/TTS.git synced 2026-09-10 11:50:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

remove debug code

Browse Source
This commit is contained in:
erogol
2020-09-11 19:01:38 +02:00
parent 77c577ddab
commit 09ad6a09b0
1 changed files with 3 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mozilla_voice_tts/bin/train_glow_tts.py
+3 -4
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']