Merge pull request adobe#40 from danielhavir/master

Fix bug in resnet
gen0924 · Sep 17, 2020 · 4f0cbd9 · 4f0cbd9
2 parents 5eb785d + c1ff7ac
commit 4f0cbd9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/antialiased_cnns/resnet.py b/antialiased_cnns/resnet.py
@@ -126,7 +126,7 @@ def __init__(self, inplanes, planes, stride=1, downsample=None, groups=1, norm_l
         # Both self.conv2 and self.downsample layers downsample the input when stride != 1
         self.conv1 = conv1x1(inplanes, planes)
         self.bn1 = norm_layer(planes)
-        self.conv2 = conv3x3(planes, planes, groups) # stride moved
+        self.conv2 = conv3x3(planes, planes, groups=groups)  # stride moved
         self.bn2 = norm_layer(planes)
         if(stride==1):
             self.conv3 = conv1x1(planes, planes * self.expansion)