From 63f7fdbf9cf2ef2401b1c8893d08c23965f4d7ec Mon Sep 17 00:00:00 2001
From: weishengyu <weishengyu@baidu.com>
Date: Wed, 8 Dec 2021 20:02:13 +0800
Subject: [PATCH] add logger

---
 ppcls/utils/gallery2fc.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/ppcls/utils/gallery2fc.py b/ppcls/utils/gallery2fc.py
index 660297f6..271527c7 100644
--- a/ppcls/utils/gallery2fc.py
+++ b/ppcls/utils/gallery2fc.py
@@ -2,6 +2,7 @@ import paddle
 from ppcls.arch import build_model
 from ppcls.utils.config import parse_config, parse_args
 from ppcls.utils.save_load import load_dygraph_pretrain
+from ppcls.utils.logger import init_logger
 
 
 def load_feature_extractor(configs):
@@ -32,6 +33,7 @@ class FuseModel(paddle.nn.Layer):
 def main():
     args = parse_args()
     configs = parse_config(args.config)
+    init_logger(name='gallery2fc')
     fuse_model = FuseModel(configs)
     save_fuse_model(fuse_model)
 
-- 
GitLab