Unverified 提交 35dc9355 authored 作者: tomguluson92's avatar tomguluson92 提交者: GitHub

remove `nn.Conv2d` in default settings (#949)

上级 c54e3946
...@@ -188,7 +188,7 @@ def parse_model(d, ch): # model_dict, input_channels(3) ...@@ -188,7 +188,7 @@ def parse_model(d, ch): # model_dict, input_channels(3)
pass pass
n = max(round(n * gd), 1) if n > 1 else n # depth gain 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] c1, c2 = ch[f], args[0]
# Normal # Normal
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论