tomguluson92 commited on
Commit
35dc935
1 Parent(s): c54e394

remove `nn.Conv2d` in default settings (#949)

Browse files
Files changed (1) hide show
  1. 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 [nn.Conv2d, Conv, Bottleneck, SPP, DWConv, MixConv2d, Focus, CrossConv, BottleneckCSP, C3]:
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