[Feature] Add groups parameter to Conv2d in ResidualBlockNoBN (#1510)

* [Enhancement] In ResidualBlockNoBN add groups parameter to Conv2d

* Update mmedit/models/common/sr_backbone_utils.py

* Update mmedit/models/common/sr_backbone_utils.py

* fix lint

---------

Co-authored-by: Yifei Yang <[email protected]>
Co-authored-by: Z-Fran <[email protected]>

Author: 3 people

mmedit/models/common/sr_backbone_utils.py

@@ -56,11 +56,13 @@ class ResidualBlockNoBN(nn.Module):
             Default: 1.0.
     """
 
-    def __init__(self, mid_channels=64, res_scale=1.0):
+    def __init__(self, mid_channels=64, res_scale=1.0, groups=1):
         super().__init__()
         self.res_scale = res_scale
-        self.conv1 = nn.Conv2d(mid_channels, mid_channels, 3, 1, 1, bias=True)
-        self.conv2 = nn.Conv2d(mid_channels, mid_channels, 3, 1, 1, bias=True)
+        self.conv1 = nn.Conv2d(
+            mid_channels, mid_channels, 3, 1, 1, bias=True, groups=groups)
+        self.conv2 = nn.Conv2d(
+            mid_channels, mid_channels, 3, 1, 1, bias=True, groups=groups)
 
         self.relu = nn.ReLU(inplace=True)