✏️ [Fix] Type hint, tuple->Tuple
Browse files- 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) ->
|
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 |
|