提交 715cb08b authored 作者: Glenn Jocher's avatar Glenn Jocher

module updates

上级 ff02ae08
...@@ -149,7 +149,7 @@ def parse_model(md, ch): # model_dict, input_channels(3) ...@@ -149,7 +149,7 @@ def parse_model(md, 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, ConvPlus, BottleneckCSP, BottleneckLight]: if m in [nn.Conv2d, Conv, Bottleneck, SPP, DWConv, MixConv2d, Focus, ConvPlus, BottleneckCSP]:
c1, c2 = ch[f], args[0] c1, c2 = ch[f], args[0]
# Normal # Normal
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论