henry000 commited on
Commit
9beb383
·
1 Parent(s): 1851849

✏️ [Fix] a type error in CBLinear

Browse files
Files changed (1) hide show
  1. 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)