fix pretty print (#1441)

* grid sample

* grid sample

* grid sample

* grid sample

* grid sample

* changelog

* version

Co-authored-by: J. Borovec <jirka.borovec@seznam.cz>
This commit is contained in:
William Falcon
2020-04-10 08:43:22 -04:00
committed by GitHub
co-authored by J. Borovec
parent 8dd9b80d7a
commit 1f685c2882
3 changed files with 5 additions and 3 deletions
+2 -2
View File
@@ -377,10 +377,10 @@ class TrainerEvaluationLoopMixin(ABC):
self.add_tqdm_metrics(prog_bar_metrics)
# log results of test
if test_mode and self.proc_rank == 0 and prog_bar_metrics:
if test_mode and self.proc_rank == 0 and len(callback_metrics) > 0:
print('-' * 80)
print('TEST RESULTS')
pprint(prog_bar_metrics)
pprint(callback_metrics)
print('-' * 80)
# log metrics