✏️ [Fix] a type error in CBLinear
Browse files- yolo/model/module.py +1 -1
yolo/model/module.py
CHANGED
@@ -236,7 +236,7 @@ class CBLinear(nn.Module):
|
|
236 |
super(CBLinear, self).__init__()
|
237 |
kwargs.setdefault("padding", auto_pad(kernel_size, **kwargs))
|
238 |
self.conv = nn.Conv2d(in_channels, sum(out_channels), kernel_size, **kwargs)
|
239 |
-
self.out_channels = out_channels
|
240 |
|
241 |
def forward(self, x: Tensor) -> Tuple[Tensor]:
|
242 |
x = self.conv(x)
|
|
|
236 |
super(CBLinear, self).__init__()
|
237 |
kwargs.setdefault("padding", auto_pad(kernel_size, **kwargs))
|
238 |
self.conv = nn.Conv2d(in_channels, sum(out_channels), kernel_size, **kwargs)
|
239 |
+
self.out_channels = list(out_channels)
|
240 |
|
241 |
def forward(self, x: Tensor) -> Tuple[Tensor]:
|
242 |
x = self.conv(x)
|