fix

zengshao0622 · Sep 27, 2022 · aee51ec · aee51ec
1 parent 23b066b
commit aee51ec
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/ppcls/arch/backbone/legendary_models/pp_lcnet.py b/ppcls/arch/backbone/legendary_models/pp_lcnet.py
@@ -302,7 +302,8 @@ def __init__(self,
         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,