Pretty test results with pprint (#1176)

This commit is contained in:
So Uchida 2020-03-25 03:52:57 +09:00 committed by GitHub
parent ce9d87597b
commit 60b8246bc3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions

View File

@ -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