module updates

This commit is contained in:
Glenn Jocher committed 2020-07-03 13:46:12 -07:00
1 parent df224a0d8f
commit 655895a838
2 files changed
+7 -17

No files matched your search

+1 -1
View File
@@ -161,7 +161,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]:
if m in [nn.Conv2d, Conv, Bottleneck, SPP, DWConv, MixConv2d, Focus, BottleneckCSP, CrossConv]:
c1, c2 = ch[f], args[0]
# Normal