🎨 [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 |
|