linter fix

This commit is contained in:
erogol 2020-06-08 10:45:38 +02:00
parent 55b8952c3d
commit f0616a3b24
3 changed files with 3 additions and 3 deletions

View File

@ -519,7 +519,7 @@ def main(args): # pylint: disable=redefined-outer-name
model_disc, criterion_disc, optimizer_disc, model_disc, criterion_disc, optimizer_disc,
scheduler_gen, scheduler_disc, ap, global_step, scheduler_gen, scheduler_disc, ap, global_step,
epoch) 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) global_step, epoch)
c_logger.print_epoch_end(epoch, eval_avg_loss_dict) c_logger.print_epoch_end(epoch, eval_avg_loss_dict)
target_loss = eval_avg_loss_dict[c.target_loss] target_loss = eval_avg_loss_dict[c.target_loss]