glenn-jocher commited on
Commit
715cb08
1 Parent(s): ff02ae0

module updates

Browse files
Files changed (1) hide show
  1. 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, BottleneckLight]:
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