From d05d9d8d6523cf7825006542bc7972ec7ff36f8a Mon Sep 17 00:00:00 2001
From: weishengyu <weishengyu@baidu.com>
Date: Wed, 8 Dec 2021 19:48:59 +0800
Subject: [PATCH] update functions

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

diff --git a/ppcls/utils/gallery2fc.py b/ppcls/utils/gallery2fc.py
index 8d9a502d6..e02a7e4f9 100644
--- a/ppcls/utils/gallery2fc.py
+++ b/ppcls/utils/gallery2fc.py
@@ -1,10 +1,6 @@
 import paddle
 from ppcls.arch import build_model
-from deploy.utils.config import parse_config, parse_args
-
-
-def load_feature_extractor(configs):
-    pass
+from ppcls.utils.config import parse_config, parse_args
 
 
 def build_gallery_feature(feature_extractor):
@@ -18,7 +14,7 @@ def save_fuse_model(fuse_model):
 class FuseModel(paddle.nn.Layer):
     def __init__(self, configs):
         super().__init__()
-        self.feature_extractor = load_feature_extractor(configs)
+        self.feature_extractor = build_model(configs)
         self.gallery_layer = build_gallery_feature(self.feature_extractor)
 
     def forward(self, x):