Glenn Jocher
4 years ago
1 changed files with 1 additions and 1 deletions
+ 1
- 1
models/yolo.py
View File
| @@ -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 | |||
Loading…