Merge pull request #1821 from WenmuZhou/dygraph_rc

mv3 small support det
2026-09-24 23:33:08 +08:00 · 2021-01-26 15:53:17 +08:00
parent b78eeb24ea b9e0a99832
commit 4dcf1b2227
1 changed files with 2 additions and 1 deletions
@@ -112,7 +112,8 @@ class MobileNetV3(nn.Layer):
        inplanes = make_divisible(inplanes * scale)
        for (k, exp, c, se, nl, s) in cfg:
            se = se and not self.disable_se
-            if s == 2 and i > 2:
+            start_idx = 2 if model_name == 'large' else 0
+            if s == 2 and i > start_idx:
                self.out_channels.append(inplanes)
                self.stages.append(nn.Sequential(*block_list))
                block_list = []