Decorate logger to display only once

ppcls/utils/__init__.py

@@ -12,10 +12,11 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from . import model_zoo
-from . import misc
 from . import logger
+from . import misc
+from . import model_zoo
 
 from .save_load import init_model, save_model
 from .config import get_config
 from .misc import AverageMeter
+

ppcls/utils/logger.py

@@ -11,63 +11,35 @@
 # 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 logging
-logging.basicConfig()
-
-DEBUG = logging.DEBUG  # 10
-INFO = logging.INFO  # 20
-WARN = logging.WARN  # 30
-ERROR = logging.ERROR  # 40
 
+import logging
+import os
 
-class Logger(object):
-    """
-    Logger
-    """
-
-    def __init__(self, level=DEBUG):
-        self.init(level)
+logging.basicConfig(level=logging.INFO)
+_logger = logging.getLogger(__name__)
 
-    def init(self, level=DEBUG):
+def anti_fleet(log):
     """
-        init
+    Because of the fucking Fleet, logs will print multi-times.
+    So we only display one of them and ignore the others.    
     """
-        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 wrapper(fmt, *args):
+        if int(os.getenv("PADDLE_TRAINER_ID", 0)) == 0:
+            log(fmt, *args)
+    return wrapper
  
+@anti_fleet
 def info(fmt, *args):
-    """info"""
     _logger.info(fmt, *args)
 
 
+@anti_fleet
 def warning(fmt, *args):
-    """warn"""
     _logger.warning(fmt, *args)
 
 
+@anti_fleet
 def error(fmt, *args):
-    """error"""
     _logger.error(fmt, *args)
 
 
@@ -86,16 +58,16 @@ def advertise():
 
     """
     copyright = "PaddleClas is powered by PaddlePaddle !"
-    info = "For more info please go to the following website."
+    ad = "For more info please go to the following website."
     website = "https://github.com/PaddlePaddle/PaddleClas"
-    AD_LEN = 6 + len(max([copyright, info, website], key=len))
+    AD_LEN = 6 + len(max([copyright, ad, website], key=len))
 
-    _logger.info("\n{0}\n{1}\n{2}\n{3}\n{4}\n{5}\n{6}\n{7}\n".format(
+    info("\n{0}\n{1}\n{2}\n{3}\n{4}\n{5}\n{6}\n{7}\n".format(
         "=" * (AD_LEN + 4),
         "=={}==".format(copyright.center(AD_LEN)),
         "=" * (AD_LEN + 4),
         "=={}==".format(' ' * AD_LEN),
-        "=={}==".format(info.center(AD_LEN)),
+        "=={}==".format(ad.center(AD_LEN)),
         "=={}==".format(' ' * AD_LEN),
         "=={}==".format(website.center(AD_LEN)),
-        "=" * (AD_LEN + 4), ))
+        "=" * (AD_LEN + 4) ))