add test handler
This commit is contained in:
@@ -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
|
||||
|
||||
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user