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

Fix for eval time logging

This commit is contained in:
Eren G 2018-07-19 16:55:00 +02:00
parent ec297bf042
commit eef059dd2f
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -283,7 +283,7 @@ def evaluate(model, criterion, criterion_st, data_loader, current_step):
# ('linear_loss', linear_loss.item()), # ('linear_loss', linear_loss.item()),
# ('mel_loss', mel_loss.item()), # ('mel_loss', mel_loss.item()),
# ('stop_loss', stop_loss.item())]) # ('stop_loss', stop_loss.item())])
if current_step % c.print_step == 0: if num_iter % c.print_step == 0:
print(" | | > TotalLoss: {:.5f} LinearLoss: {:.5f} MelLoss:{:.5f} "\ print(" | | > TotalLoss: {:.5f} LinearLoss: {:.5f} MelLoss:{:.5f} "\
"StopLoss: {:.5f} ".format(loss.item(), "StopLoss: {:.5f} ".format(loss.item(),
linear_loss.item(), linear_loss.item(),