linter fix

2020-06-08 10:45:38 +02:00 · 2020-06-08 10:45:38 +02:00 · dbe381a186
parent 8d307f2133
commit dbe381a186
3 changed files with 3 additions and 3 deletions
--- a/vocoder/train.py
+++ b/vocoder/train.py
@ -519,7 +519,7 @@ def main(args):  # pylint: disable=redefined-outer-name
                               model_disc, criterion_disc, optimizer_disc,
                               scheduler_gen, scheduler_disc, ap, global_step,
                               epoch)
-        eval_avg_loss_dict = evaluate(model_gen, criterion_gen, model_disc, ap,
+        eval_avg_loss_dict = evaluate(model_gen, criterion_gen, model_disc, criterion_disc, ap,
                                      global_step, epoch)
        c_logger.print_epoch_end(epoch, eval_avg_loss_dict)
        target_loss = eval_avg_loss_dict[c.target_loss]