From c5969f79f535d9b0a6627c3f6a0e40e0f4e15888 Mon Sep 17 00:00:00 2001
From: tkianai <49858938+tkianai@users.noreply.github.com>
Date: Wed, 2 Sep 2020 13:18:42 +0800
Subject: [PATCH] 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
---
 utils/torch_utils.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/utils/torch_utils.py b/utils/torch_utils.py
index c024983..bc54ab1 100644
--- a/utils/torch_utils.py
+++ b/utils/torch_utils.py
@@ -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