Unverified 提交 c5969f79 authored 作者: tkianai's avatar tkianai 提交者: GitHub

add support for DWConv fuse operation (#898)

* the value should be a list to unpacking * filter the None labels * update readme to remove apex dependency * add support for DWConv fuse operation
上级 d3f9bf2b
...@@ -112,6 +112,7 @@ def fuse_conv_and_bn(conv, bn): ...@@ -112,6 +112,7 @@ def fuse_conv_and_bn(conv, bn):
kernel_size=conv.kernel_size, kernel_size=conv.kernel_size,
stride=conv.stride, stride=conv.stride,
padding=conv.padding, padding=conv.padding,
groups=conv.groups,
bias=True).to(conv.weight.device) bias=True).to(conv.weight.device)
# prepare filters # prepare filters
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论