module updates
This commit is contained in:
1 parent
ff02ae0869
commit
715cb08b10
1 file changed
+1
-1
+1
-1
@@ -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
|
||||
|
||||
Reference in new issue
Block a user