Spaces:
Sleeping
Sleeping
File size: 1,128 Bytes
4a1df2e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 |
import logging
import os
import sys
import time
import termcolor
today = time.strftime("%Y%m%d %H%M%S", time.localtime(time.time()))
def get_logger(log_path, log_name="", log_type="training_log"):
if not log_path:
log_dir = os.path.join(log_path, "logs")
else:
log_dir = os.path.join(".", "logs")
full_path = os.path.join(log_dir, log_name + "_" + today)
if not os.path.exists(full_path):
os.makedirs(full_path)
log_path = os.path.join(full_path, "{}.log".format(log_type))
logger = logging.getLogger(log_name)
if not logger.handlers:
formatter = logging.Formatter("%(asctime)s %(levelname)s: %(message)s")
file_handler = logging.FileHandler(log_path, encoding="utf8")
file_handler.setFormatter(formatter)
file_handler.setLevel(logging.INFO)
console_handler = logging.StreamHandler(sys.stdout)
console_handler.formatter = formatter
console_handler.setLevel(logging.INFO)
logger.addHandler(file_handler)
logger.addHandler(console_handler)
logger.setLevel(logging.INFO)
return logger
|