提交 aee51ece 编写于 作者: G gaotingquan 提交者: Tingquan Gao

fix

上级 23b066be
......@@ -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,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册