From f29bb526fc52f25d9d26195d16257ccb3c9fb6b4 Mon Sep 17 00:00:00 2001 From: WuHaobo Date: Sat, 11 Apr 2020 02:35:22 +0800 Subject: [PATCH] add the logger --- .gitignore | 5 ++- ppcls/utils/logger.py | 73 +++++++++++++++++++++++++++++++++++++++++++ 2 files changed, 75 insertions(+), 3 deletions(-) create mode 100644 ppcls/utils/logger.py diff --git a/.gitignore b/.gitignore index 6062e298..53c72a69 100644 --- a/.gitignore +++ b/.gitignore @@ -1,8 +1,7 @@ -!ppcls/ *.pyc *.sw* -*log* -/dataset +*/workerlog* +dataset/ checkpoints/ output/ pretrained/ diff --git a/ppcls/utils/logger.py b/ppcls/utils/logger.py new file mode 100644 index 00000000..bcc8ebe6 --- /dev/null +++ b/ppcls/utils/logger.py @@ -0,0 +1,73 @@ +#copyright (c) 2020 PaddlePaddle Authors. All Rights Reserve. +# +#Licensed under the Apache License, Version 2.0 (the "License"); +#you may not use this file except in compliance with the License. +#You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +#Unless required by applicable law or agreed to in writing, software +#distributed under the License is distributed on an "AS IS" BASIS, +#WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +#See the License for the specific language governing permissions and +#limitations under the License. + +import os +import logging +import random + +DEBUG = logging.DEBUG #10 +INFO = logging.INFO #20 +WARN = logging.WARN #30 +ERROR = logging.ERROR #40 + + +class Logger(object): + """ + Logger + """ + + def __init__(self, level=DEBUG): + self.init(level) + + def init(self, level=DEBUG): + """ + init + """ + self._logger = logging.getLogger() + self._logger.setLevel(level) + + def info(self, fmt, *args): + """info""" + self._logger.info(fmt, *args) + + def warning(self, fmt, *args): + """warning""" + self._logger.warning(fmt, *args) + + def error(self, fmt, *args): + """error""" + self._logger.error(fmt, *args) + + +_logger = Logger() + + +def init(level=DEBUG): + """init for external""" + _logger.init(level) + + +def info(fmt, *args): + """info""" + _logger.info(fmt, *args) + + +def warning(fmt, *args): + """warn""" + _logger.warning(fmt, *args) + + +def error(fmt, *args): + """error""" + _logger.error(fmt, *args) -- GitLab