提交 ec36dd90 编写于 作者: G gaotingquan

fix: init logger for building model with loading pretrained weights

上级 0efcf0b6
...@@ -40,6 +40,9 @@ from deploy.utils import config ...@@ -40,6 +40,9 @@ from deploy.utils import config
from ppcls.arch.backbone import * from ppcls.arch.backbone import *
from ppcls.utils.logger import init_logger from ppcls.utils.logger import init_logger
# for building model with loading pretrained weights from backbone
init_logger()
__all__ = ["PaddleClas"] __all__ = ["PaddleClas"]
BASE_DIR = os.path.expanduser("~/.paddleclas/") BASE_DIR = os.path.expanduser("~/.paddleclas/")
...@@ -410,7 +413,6 @@ class PaddleClas(object): ...@@ -410,7 +413,6 @@ class PaddleClas(object):
"""PaddleClas. """PaddleClas.
""" """
init_logger(name='root')
print_info() print_info()
def __init__(self, def __init__(self,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册