def _setup_logger(logger, logdir: str):
ch = logging.StreamHandler(sys.stdout)
ch.setLevel(logging.INFO)
txt_formatter = formatters.TxtMetricsFormatter()
ch.setFormatter(txt_formatter)
# add the handlers to the logger
logger.addHandler(ch)
if logdir:
fh = logging.FileHandler(f"{logdir}/log.txt")
fh.setLevel(logging.INFO)
fh.setFormatter(txt_formatter)
logger.addHandler(fh)
def _get_logger(logdir):
logger = logging.getLogger("metrics_logger")
logger.setLevel(logging.INFO)
ch = logging.StreamHandler(sys.stdout)
ch.setLevel(logging.INFO)
txt_formatter = formatters.TxtMetricsFormatter()
ch.setFormatter(txt_formatter)
# add the handlers to the logger
logger.addHandler(ch)
if logdir:
fh = logging.FileHandler(f"{logdir}/log.txt")
fh.setLevel(logging.INFO)
fh.setFormatter(txt_formatter)
logger.addHandler(fh)
# logger.addHandler(jh)
return logger