tomguluson92
commited on
Commit
•
35dc935
1
Parent(s):
c54e394
remove `nn.Conv2d` in default settings (#949)
Browse files- models/yolo.py +1 -1
models/yolo.py
CHANGED
@@ -188,7 +188,7 @@ def parse_model(d, ch): # model_dict, input_channels(3)
|
|
188 |
pass
|
189 |
|
190 |
n = max(round(n * gd), 1) if n > 1 else n # depth gain
|
191 |
-
if m in [
|
192 |
c1, c2 = ch[f], args[0]
|
193 |
|
194 |
# Normal
|
|
|
188 |
pass
|
189 |
|
190 |
n = max(round(n * gd), 1) if n > 1 else n # depth gain
|
191 |
+
if m in [Conv, Bottleneck, SPP, DWConv, MixConv2d, Focus, CrossConv, BottleneckCSP, C3]:
|
192 |
c1, c2 = ch[f], args[0]
|
193 |
|
194 |
# Normal
|