mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-10-29 13:30:57 +08:00
Pretty test results with pprint (#1176)
This commit is contained in:
1 parent
ce9d87597b
commit
60b8246bc3
1 file changed
+2
-1
@@ -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
|
||||
|
||||
Reference in new issue
Block a user