henry000 commited on
Commit
7dd7b62
·
1 Parent(s): b35b9ac

🎨 [Update] module with formatter, fix CI error

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