From 72fe5a8217f1e28439aec902f72bf8eaff7a7521 Mon Sep 17 00:00:00 2001 From: erogol Date: Fri, 5 Jun 2020 13:27:09 +0200 Subject: [PATCH] correct name --- vocoder/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vocoder/train.py b/vocoder/train.py index 82f3bee6..2c3a2e0a 100644 --- a/vocoder/train.py +++ b/vocoder/train.py @@ -363,7 +363,7 @@ def evaluate(model_G, criterion_G, model_D, ap, global_step, epoch): for key, value in loss_G_dict.items(): update_eval_values['avg_' + key] = value.item() update_eval_values['avg_loader_time'] = loader_time - update_eval_values['avgP_step_time'] = step_time + update_eval_values['avg_step_time'] = step_time keep_avg.update_values(update_eval_values) # print eval stats