From 60b8246bc3d4e22f806780a816878c668ac647c4 Mon Sep 17 00:00:00 2001 From: So Uchida Date: Wed, 25 Mar 2020 03:52:57 +0900 Subject: [PATCH] Pretty test results with pprint (#1176) --- pytorch_lightning/trainer/evaluation_loop.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/pytorch_lightning/trainer/evaluation_loop.py b/pytorch_lightning/trainer/evaluation_loop.py index d7a2e1799d..bfbca4c25d 100644 --- a/pytorch_lightning/trainer/evaluation_loop.py +++ b/pytorch_lightning/trainer/evaluation_loop.py @@ -126,6 +126,7 @@ In this second case, the options you pass to trainer will be used when running import sys import warnings from abc import ABC, abstractmethod +from pprint import pprint from typing import Callable import torch @@ -376,7 +377,7 @@ class TrainerEvaluationLoopMixin(ABC): if self.proc_rank == 0: print('-' * 100) print('TEST RESULTS') - print(prog_bar_metrics) + pprint(prog_bar_metrics) print('-' * 100) # log metrics