henry000 commited on
Commit
930952c
·
1 Parent(s): 31c2c1a

✏️ [Fix] Type hint, tuple->Tuple

Browse files
Files changed (1) hide show
  1. yolo/model/module.py +2 -2
yolo/model/module.py CHANGED
@@ -1,4 +1,4 @@
1
- from typing import Optional
2
 
3
  import torch
4
  from torch import Tensor, nn
@@ -67,7 +67,7 @@ class CBLinear(nn.Module):
67
  self.conv = nn.Conv2d(in_channels, sum(out_channels), kernel_size, **kwargs)
68
  self.out_channels = out_channels
69
 
70
- def forward(self, x: Tensor) -> tuple[Tensor]:
71
  x = self.conv(x)
72
  return x.split(self.out_channels, dim=1)
73
 
 
1
+ from typing import Optional, Tuple
2
 
3
  import torch
4
  from torch import Tensor, nn
 
67
  self.conv = nn.Conv2d(in_channels, sum(out_channels), kernel_size, **kwargs)
68
  self.out_channels = out_channels
69
 
70
+ def forward(self, x: Tensor) -> Tuple[Tensor]:
71
  x = self.conv(x)
72
  return x.split(self.out_channels, dim=1)
73