fix RM save model

Former-commit-id: 8104cc2425431eb1cddccf3909855296116f922b
This commit is contained in:
hiyouga
2023-08-01 11:56:17 +08:00
parent 9bba01a033
commit 8e26eb374e
7 changed files with 33 additions and 16 deletions

View File

@@ -16,8 +16,16 @@ class LoggerHandler(logging.Handler):
self.log += "\n\n"
def get_logger(name: str) -> logging.Logger:
def reset_logging():
r"""
Removes basic config of root logger
"""
root = logging.getLogger()
list(map(root.removeHandler, root.handlers))
list(map(root.removeFilter, root.filters))
def get_logger(name: str) -> logging.Logger:
formatter = logging.Formatter(
fmt="%(asctime)s - %(levelname)s - %(name)s - %(message)s",
datefmt="%m/%d/%Y %H:%M:%S"