tkianai commited on
Commit
c5969f7
·
unverified ·
1 Parent(s): d3f9bf2

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

Files changed (1) hide show
  1. 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