提交 63f7fdbf 编写于 作者: W weishengyu

add logger

上级 823074d8
...@@ -2,6 +2,7 @@ import paddle ...@@ -2,6 +2,7 @@ import paddle
from ppcls.arch import build_model from ppcls.arch import build_model
from ppcls.utils.config import parse_config, parse_args from ppcls.utils.config import parse_config, parse_args
from ppcls.utils.save_load import load_dygraph_pretrain from ppcls.utils.save_load import load_dygraph_pretrain
from ppcls.utils.logger import init_logger
def load_feature_extractor(configs): def load_feature_extractor(configs):
...@@ -32,6 +33,7 @@ class FuseModel(paddle.nn.Layer): ...@@ -32,6 +33,7 @@ class FuseModel(paddle.nn.Layer):
def main(): def main():
args = parse_args() args = parse_args()
configs = parse_config(args.config) configs = parse_config(args.config)
init_logger(name='gallery2fc')
fuse_model = FuseModel(configs) fuse_model = FuseModel(configs)
save_fuse_model(fuse_model) save_fuse_model(fuse_model)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册