diff --git a/layers/BBB/BBBConv.py b/layers/BBB/BBBConv.py index 1e901037..d8b39c40 100755 --- 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