We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
2 parents 6ece34b + 23e84ac commit d93bad5Copy full SHA for d93bad5
1 file changed
layers/BBB/BBBConv.py
@@ -18,7 +18,7 @@ def __init__(self, in_channels, out_channels, kernel_size,
18
super(BBBConv2d, self).__init__()
19
self.in_channels = in_channels
20
self.out_channels = out_channels
21
- self.kernel_size = (kernel_size, kernel_size)
+ self.kernel_size = kernel_size if isinstance(kernel_size, tuple) else (kernel_size, kernel_size)
22
self.stride = stride
23
self.padding = padding
24
self.dilation = dilation
0 commit comments