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:
1 parent
d3f9bf2bb7
commit
c5969f79f5
1 file changed
+1
@@ -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
|
||||
|
||||
Reference in new issue
Block a user