add test handler

This commit is contained in:
2020-08-24 22:46:36 +08:00
parent 31aafb3470
commit 56b355737f
5 changed files with 65 additions and 9 deletions

View File

@@ -17,7 +17,7 @@ def empty_cuda_cache(_):
def setup_common_handlers(trainer: Engine, config, stop_on_nan=True, clear_cuda_cache=False, use_profiler=True,
to_save=None, end_event=None, set_epoch_for_dist_sampler=True):
to_save=None, end_event=None, set_epoch_for_dist_sampler=False):
"""
Helper method to setup trainer with common handlers.
1. TerminateOnNan

View File

@@ -1,5 +1,6 @@
import logging
from typing import Optional
from pathlib import Path
def setup_logger(
@@ -76,10 +77,12 @@ def setup_logger(
ch.setFormatter(formatter)
logger.addHandler(ch)
if filepath is not None:
if filepath is not None and Path(filepath).parent.exists():
fh = logging.FileHandler(filepath)
fh.setLevel(logging.DEBUG)
fh.setFormatter(formatter)
logger.addHandler(fh)
else:
logger.warning("not set file logger")
return logger