tkianai
commited on
add support for DWConv fuse operation (#898)
Browse files* the value should be a list to unpacking
* filter the None labels
* update readme to remove apex dependency
* add support for DWConv fuse operation
- utils/torch_utils.py +1 -0
utils/torch_utils.py
CHANGED
@@ -112,6 +112,7 @@ def fuse_conv_and_bn(conv, bn):
|
|
112 |
kernel_size=conv.kernel_size,
|
113 |
stride=conv.stride,
|
114 |
padding=conv.padding,
|
|
|
115 |
bias=True).to(conv.weight.device)
|
116 |
|
117 |
# prepare filters
|
|
|
112 |
kernel_size=conv.kernel_size,
|
113 |
stride=conv.stride,
|
114 |
padding=conv.padding,
|
115 |
+
groups=conv.groups,
|
116 |
bias=True).to(conv.weight.device)
|
117 |
|
118 |
# prepare filters
|