diff --git a/core/trainers/framework/runner.py b/core/trainers/framework/runner.py index baab7315033eff11f0b6be601b4757ed2e3707c9..761c6a748d9875f12ff48fb68da4822690fe8790 100644 --- a/core/trainers/framework/runner.py +++ b/core/trainers/framework/runner.py @@ -152,6 +152,7 @@ class RunnerBase(object): metrics_format.append("{}: {{:.2f}}s".format("time_each_interval")) metrics_names = ["total_batch"] + for name, var in metrics.items(): metrics_names.append(name) metrics_varnames.append(var.name)