diff --git a/ppcls/arch/backbone/legendary_models/pp_lcnet.py b/ppcls/arch/backbone/legendary_models/pp_lcnet.py index 017592d3f2cbfedf9bc64f3827c14790ff2c0ed0..bb9dc6c767d46aa748bee6234889d8607bcaf1ee 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,