提交 9acc8a75 编写于 作者: W WuHaobo

Decorate logger to display only once

上级 8885274f
...@@ -19,4 +19,3 @@ from . import model_zoo ...@@ -19,4 +19,3 @@ from . import model_zoo
from .save_load import init_model, save_model from .save_load import init_model, save_model
from .config import get_config from .config import get_config
from .misc import AverageMeter from .misc import AverageMeter
...@@ -18,16 +18,20 @@ import os ...@@ -18,16 +18,20 @@ import os
logging.basicConfig(level=logging.INFO) logging.basicConfig(level=logging.INFO)
_logger = logging.getLogger(__name__) _logger = logging.getLogger(__name__)
def anti_fleet(log): def anti_fleet(log):
""" """
Because of the fucking Fleet, logs will print multi-times. Because of the fucking Fleet, logs will print multi-times.
So we only display one of them and ignore the others. So we only display one of them and ignore the others.
""" """
def wrapper(fmt, *args): def wrapper(fmt, *args):
if int(os.getenv("PADDLE_TRAINER_ID", 0)) == 0: if int(os.getenv("PADDLE_TRAINER_ID", 0)) == 0:
log(fmt, *args) log(fmt, *args)
return wrapper return wrapper
@anti_fleet @anti_fleet
def info(fmt, *args): def info(fmt, *args):
_logger.info(fmt, *args) _logger.info(fmt, *args)
...@@ -70,4 +74,4 @@ def advertise(): ...@@ -70,4 +74,4 @@ def advertise():
"=={}==".format(ad.center(AD_LEN)), "=={}==".format(ad.center(AD_LEN)),
"=={}==".format(' ' * AD_LEN), "=={}==".format(' ' * AD_LEN),
"=={}==".format(website.center(AD_LEN)), "=={}==".format(website.center(AD_LEN)),
"=" * (AD_LEN + 4) )) "=" * (AD_LEN + 4), ))
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册