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
This commit is contained in:
tkianai authored and GitHub committed 2020-09-01 22:18:42 -07:00
1 parent d3f9bf2bb7
commit c5969f79f5
1 file changed
+1
+1
View File
@@ -112,6 +112,7 @@ def fuse_conv_and_bn(conv, bn):
kernel_size=conv.kernel_size,
stride=conv.stride,
padding=conv.padding,
groups=conv.groups,
bias=True).to(conv.weight.device)
# prepare filters