From b662ed34acde16995d869fe3bc996f91b9071881 Mon Sep 17 00:00:00 2001
From: weishengyu <weishengyu@baidu.com>
Date: Fri, 24 Dec 2021 11:15:27 +0800
Subject: [PATCH] dbg
---
ppcls/utils/gallery2fc.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/ppcls/utils/gallery2fc.py b/ppcls/utils/gallery2fc.py
index cc626cdcf..4445feb56 100644
--- a/ppcls/utils/gallery2fc.py
+++ b/ppcls/utils/gallery2fc.py
@@ -40,7 +40,7 @@ class GalleryLayer(paddle.nn.Layer):
gallery_labels.append(line[1].strip())
self.gallery_layer = paddle.nn.Linear(embedding_size, len(self.gallery_images), bias_attr=False)
- def forward(self, x):
+ def forward(self, x, label=None):
x = paddle.nn.functional.normalize(x)
x = self.gallery_layer(x)
return x
@@ -72,7 +72,7 @@ def export_fuse_model(configs):
configs["Slim"] = None
fuse_model = build_model(configs)
fuse_model.head = GalleryLayer(configs)
- configs["slim"] = slim_config
+ configs["Slim"] = slim_config
quantize_model(configs, fuse_model)
load_dygraph_pretrain(fuse_model, configs["Global"]["pretrained_model"])
fuse_model.eval()