Enable Logging

This commit is contained in:
XXXXRT666 2025-05-14 08:50:01 +01:00
parent cbf9a9bdbd
commit b310dadf8c

View File

@ -16,7 +16,7 @@ logging.getLogger("matplotlib").setLevel(logging.ERROR)
MATPLOTLIB_FLAG = False
logging.basicConfig(stream=sys.stdout, level=logging.ERROR)
logging.basicConfig(stream=sys.stdout, level=logging.INFO)
logger = logging
@ -309,13 +309,13 @@ def check_git_hash(model_dir):
def get_logger(model_dir, filename="train.log"):
global logger
logger = logging.getLogger(os.path.basename(model_dir))
logger.setLevel(logging.ERROR)
logger.setLevel(logging.INFO)
formatter = logging.Formatter("%(asctime)s\t%(name)s\t%(levelname)s\t%(message)s")
if not os.path.exists(model_dir):
os.makedirs(model_dir)
h = logging.FileHandler(os.path.join(model_dir, filename))
h.setLevel(logging.ERROR)
h.setLevel(logging.INFO)
h.setFormatter(formatter)
logger.addHandler(h)
return logger