Spaces:
Running
Running
🎨 [Update] module with formatter, fix CI error
Browse files- yolo/model/module.py +5 -5
yolo/model/module.py
CHANGED
|
@@ -20,7 +20,7 @@ class Conv(nn.Module):
|
|
| 20 |
kernel_size: _size_2_t,
|
| 21 |
*,
|
| 22 |
activation: Optional[str] = "SiLU",
|
| 23 |
-
**kwargs
|
| 24 |
):
|
| 25 |
super().__init__()
|
| 26 |
kwargs.setdefault("padding", auto_pad(kernel_size, **kwargs))
|
|
@@ -106,7 +106,7 @@ class RepConv(nn.Module):
|
|
| 106 |
kernel_size: _size_2_t = 3,
|
| 107 |
*,
|
| 108 |
activation: Optional[str] = "SiLU",
|
| 109 |
-
**kwargs
|
| 110 |
):
|
| 111 |
super().__init__()
|
| 112 |
self.act = create_activation_function(activation)
|
|
@@ -128,7 +128,7 @@ class RepNBottleneck(nn.Module):
|
|
| 128 |
kernel_size: Tuple[int, int] = (3, 3),
|
| 129 |
residual: bool = True,
|
| 130 |
expand: float = 1.0,
|
| 131 |
-
**kwargs
|
| 132 |
):
|
| 133 |
super().__init__()
|
| 134 |
neck_channels = int(out_channels * expand)
|
|
@@ -159,7 +159,7 @@ class RepNCSP(nn.Module):
|
|
| 159 |
csp_expand: float = 0.5,
|
| 160 |
repeat_num: int = 1,
|
| 161 |
neck_args: Dict[str, Any] = {},
|
| 162 |
-
**kwargs
|
| 163 |
):
|
| 164 |
super().__init__()
|
| 165 |
|
|
@@ -190,7 +190,7 @@ class RepNCSPELAN(nn.Module):
|
|
| 190 |
process_channels: Optional[int] = None,
|
| 191 |
csp_args: Dict[str, Any] = {},
|
| 192 |
csp_neck_args: Dict[str, Any] = {},
|
| 193 |
-
**kwargs
|
| 194 |
):
|
| 195 |
super().__init__()
|
| 196 |
|
|
|
|
| 20 |
kernel_size: _size_2_t,
|
| 21 |
*,
|
| 22 |
activation: Optional[str] = "SiLU",
|
| 23 |
+
**kwargs,
|
| 24 |
):
|
| 25 |
super().__init__()
|
| 26 |
kwargs.setdefault("padding", auto_pad(kernel_size, **kwargs))
|
|
|
|
| 106 |
kernel_size: _size_2_t = 3,
|
| 107 |
*,
|
| 108 |
activation: Optional[str] = "SiLU",
|
| 109 |
+
**kwargs,
|
| 110 |
):
|
| 111 |
super().__init__()
|
| 112 |
self.act = create_activation_function(activation)
|
|
|
|
| 128 |
kernel_size: Tuple[int, int] = (3, 3),
|
| 129 |
residual: bool = True,
|
| 130 |
expand: float = 1.0,
|
| 131 |
+
**kwargs,
|
| 132 |
):
|
| 133 |
super().__init__()
|
| 134 |
neck_channels = int(out_channels * expand)
|
|
|
|
| 159 |
csp_expand: float = 0.5,
|
| 160 |
repeat_num: int = 1,
|
| 161 |
neck_args: Dict[str, Any] = {},
|
| 162 |
+
**kwargs,
|
| 163 |
):
|
| 164 |
super().__init__()
|
| 165 |
|
|
|
|
| 190 |
process_channels: Optional[int] = None,
|
| 191 |
csp_args: Dict[str, Any] = {},
|
| 192 |
csp_neck_args: Dict[str, Any] = {},
|
| 193 |
+
**kwargs,
|
| 194 |
):
|
| 195 |
super().__init__()
|
| 196 |
|