glenn-jocher
commited on
Commit
•
715cb08
1
Parent(s):
ff02ae0
module updates
Browse files- models/yolo.py +1 -1
models/yolo.py
CHANGED
@@ -149,7 +149,7 @@ def parse_model(md, ch): # model_dict, input_channels(3)
|
|
149 |
pass
|
150 |
|
151 |
n = max(round(n * gd), 1) if n > 1 else n # depth gain
|
152 |
-
if m in [nn.Conv2d, Conv, Bottleneck, SPP, DWConv, MixConv2d, Focus, ConvPlus, BottleneckCSP
|
153 |
c1, c2 = ch[f], args[0]
|
154 |
|
155 |
# Normal
|
|
|
149 |
pass
|
150 |
|
151 |
n = max(round(n * gd), 1) if n > 1 else n # depth gain
|
152 |
+
if m in [nn.Conv2d, Conv, Bottleneck, SPP, DWConv, MixConv2d, Focus, ConvPlus, BottleneckCSP]:
|
153 |
c1, c2 = ch[f], args[0]
|
154 |
|
155 |
# Normal
|