add export method

ppcls/utils/gallery2fc.py

@@ -57,18 +57,6 @@ def build_gallery_layer(configs, feature_extractor):
     return gallery_layer
 
 
-def export_fuse_model(model, config):
-    model.eval()
-    model.quanter.save_quantized_model(
-        model.base_model,
-        save_path,
-        input_spec=[
-            paddle.static.InputSpec(
-                shape=[None] + config["Global"]["image_shape"],
-                dtype='float32')
-        ])
-
-
 class FuseModel(paddle.nn.Layer):
     def __init__(self, configs):
         super().__init__()
@@ -85,12 +73,25 @@ class FuseModel(paddle.nn.Layer):
         return x
 
 
+def export_fuse_model(configs):
+    fuse_model = FuseModel(configs)
+    fuse_model.eval()
+    save_path = configs["Global"]["save_inference_dir"]
+    fuse_model.quanter.save_quantized_model(
+        fuse_model,
+        save_path,
+        input_spec=[
+            paddle.static.InputSpec(
+                shape=[None] + configs["Global"]["image_shape"],
+                dtype='float32')
+        ])
+
+
 def main():
     args = parse_args()
     configs = parse_config(args.config)
     init_logger(name='gallery2fc')
-    fuse_model = FuseModel(configs)
-    # save_fuse_model(fuse_model)
+    export_fuse_model(configs)
 
 
 if __name__ == '__main__':