Fix for eval time logging

2026-09-07 16:40:44 +08:00 · 2018-07-19 16:55:00 +02:00
parent ec297bf042
commit eef059dd2f
1 changed files with 1 additions and 1 deletions
@@ -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(),