Merge pull request #60 from mttgdd/master

allow for non-square kernels
kumar-shridhar · Feb 5, 2021 · d93bad5 · d93bad5
2 parents 6ece34b + 23e84ac
commit d93bad5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/layers/BBB/BBBConv.py b/layers/BBB/BBBConv.py
@@ -18,7 +18,7 @@ def __init__(self, in_channels, out_channels, kernel_size,
         super(BBBConv2d, self).__init__()
         self.in_channels = in_channels
         self.out_channels = out_channels
-        self.kernel_size = (kernel_size, kernel_size)
+        self.kernel_size = kernel_size if isinstance(kernel_size, tuple) else (kernel_size, kernel_size)
         self.stride = stride
         self.padding = padding
         self.dilation = dilation