diff --git a/ppcls/utils/gallery2fc.py b/ppcls/utils/gallery2fc.py index 660297f663fe9a1eb709308d6837dd5837bac038..271527c78288db4e1811e3b8ff137ea947c49770 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)