remove nn.Conv2d in default settings (#949)

This commit is contained in:
tomguluson92 authored and GitHub committed 2020-09-11 16:35:48 -07:00
1 parent c54e39464c
commit 35dc935504
1 file changed
+1 -1
+1 -1
View File
@@ -188,7 +188,7 @@ def parse_model(d, 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, CrossConv, BottleneckCSP, C3]:
if m in [Conv, Bottleneck, SPP, DWConv, MixConv2d, Focus, CrossConv, BottleneckCSP, C3]:
c1, c2 = ch[f], args[0]
# Normal