From 7dd7b628a945e37ec494b3278f432c194da1aa96 Mon Sep 17 00:00:00 2001 From: henrytsui000 Date: Tue, 4 Jun 2024 13:10:39 +0800 Subject: [PATCH] =?UTF-8?q?=F0=9F=8E=A8=20[Update]=20module=20with=20forma?= =?UTF-8?q?tter,=20fix=20CI=20error?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- yolo/model/module.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/yolo/model/module.py b/yolo/model/module.py index df9e03b..30002e4 100644 --- a/yolo/model/module.py +++ b/yolo/model/module.py @@ -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)) @@ -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) @@ -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) @@ -159,7 +159,7 @@ def __init__( csp_expand: float = 0.5, repeat_num: int = 1, neck_args: Dict[str, Any] = {}, - **kwargs + **kwargs, ): super().__init__() @@ -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__()