Skip to content

Commit

Permalink
🎨 [Update] module with formatter, fix CI error
Browse files Browse the repository at this point in the history
  • Loading branch information
henrytsui000 committed Jun 4, 2024
1 parent b35b9ac commit 7dd7b62
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions yolo/model/module.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,7 @@ def __init__(
kernel_size: _size_2_t,
*,
activation: Optional[str] = "SiLU",
**kwargs
**kwargs,
):
super().__init__()
kwargs.setdefault("padding", auto_pad(kernel_size, **kwargs))
Expand Down Expand Up @@ -106,7 +106,7 @@ def __init__(
kernel_size: _size_2_t = 3,
*,
activation: Optional[str] = "SiLU",
**kwargs
**kwargs,
):
super().__init__()
self.act = create_activation_function(activation)
Expand All @@ -128,7 +128,7 @@ def __init__(
kernel_size: Tuple[int, int] = (3, 3),
residual: bool = True,
expand: float = 1.0,
**kwargs
**kwargs,
):
super().__init__()
neck_channels = int(out_channels * expand)
Expand Down Expand Up @@ -159,7 +159,7 @@ def __init__(
csp_expand: float = 0.5,
repeat_num: int = 1,
neck_args: Dict[str, Any] = {},
**kwargs
**kwargs,
):
super().__init__()

Expand Down Expand Up @@ -190,7 +190,7 @@ def __init__(
process_channels: Optional[int] = None,
csp_args: Dict[str, Any] = {},
csp_neck_args: Dict[str, Any] = {},
**kwargs
**kwargs,
):
super().__init__()

Expand Down

0 comments on commit 7dd7b62

Please sign in to comment.