From 0160b8dd0d4790ac43fdfe753b08dedefb567308 Mon Sep 17 00:00:00 2001
From: gaotingquan <gaotingquan@baidu.com>
Date: Tue, 27 Sep 2022 02:56:31 +0000
Subject: [PATCH] fix

---
 ppcls/arch/backbone/legendary_models/pp_lcnet.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/ppcls/arch/backbone/legendary_models/pp_lcnet.py b/ppcls/arch/backbone/legendary_models/pp_lcnet.py
index 017592d3..bb9dc6c7 100644
--- a/ppcls/arch/backbone/legendary_models/pp_lcnet.py
+++ b/ppcls/arch/backbone/legendary_models/pp_lcnet.py
@@ -302,7 +302,8 @@ class PPLCNet(TheseusLayer):
         self.flatten = nn.Flatten(start_axis=1, stop_axis=-1)
         self.fc = Linear(
             self.class_expand if self.use_last_conv else
-            make_divisible(self.net_config["blocks6"][-1][2]), class_num)
+            make_divisible(self.net_config["blocks6"][-1][2] * scale),
+            class_num)
 
         super().init_res(
             stages_pattern,
-- 
GitLab