Merge pull request #506 from ALECX123/main

fix issue #156 UnicodeEncodeError
This commit is contained in:
Songting
2023-10-28 14:36:10 +08:00
committed by GitHub
+1 -1
View File
@@ -395,7 +395,7 @@ def get_logger(model_dir, filename="train.log"):
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 = logging.FileHandler(os.path.join(model_dir, filename),encoding="utf-8")
h.setLevel(logging.DEBUG)
h.setFormatter(formatter)
logger.addHandler(h)