module updates

5 years ago · 715cb08b10
parent ff02ae0869
commit 715cb08b10
1 changed files with 1 additions and 1 deletions
--- a/models/yolo.py
+++ b/models/yolo.py
@ -149,7 +149,7 @@ def parse_model(md, ch):  # model_dict, input_channels(3)
                pass

        n = max(round(n * gd), 1) if n > 1 else n  # depth gain
-        if m in [nn.Conv2d, Conv, Bottleneck, SPP, DWConv, MixConv2d, Focus, ConvPlus, BottleneckCSP, BottleneckLight]:
+        if m in [nn.Conv2d, Conv, Bottleneck, SPP, DWConv, MixConv2d, Focus, ConvPlus, BottleneckCSP]:
            c1, c2 = ch[f], args[0]

            # Normal