mirror of
https://github.com/wassname/TTS.git
synced 2026-10-25 12:10:21 +08:00
Fix for eval time logging
This commit is contained in:
1 parent
ec297bf042
commit
eef059dd2f
1 file changed
+1
-1
@@ -283,7 +283,7 @@ def evaluate(model, criterion, criterion_st, data_loader, current_step):
|
||||
# ('linear_loss', linear_loss.item()),
|
||||
# ('mel_loss', mel_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} "\
|
||||
"StopLoss: {:.5f} ".format(loss.item(),
|
||||
linear_loss.item(),
|
||||
|
||||
Reference in new issue
Block a user