From 9e4c519f11b5a975c9fe47fdb3d247b0767fafc0 Mon Sep 17 00:00:00 2001 From: leondgarse Date: Sat, 19 Aug 2023 18:31:59 +0800 Subject: [PATCH] Update replknet.py --- models/modules/replknet.py | 5 ----- 1 file changed, 5 deletions(-) diff --git a/models/modules/replknet.py b/models/modules/replknet.py index 2af0157..46dd29b 100644 --- a/models/modules/replknet.py +++ b/models/modules/replknet.py @@ -28,7 +28,6 @@ def __init__( groups: int, small_kernel: int, inference_mode: bool = False, - activation: nn.Module = nn.GELU(), ) -> None: """Construct a ReparamLargeKernelConv module. @@ -40,7 +39,6 @@ def __init__( groups: Group number. Default: 1 small_kernel: Kernel size of small kernel conv branch. inference_mode: If True, instantiates model in inference mode. Default: ``False`` - activation: Activation module. Default: ``nn.GELU`` """ super(ReparamLargeKernelConv, self).__init__() @@ -48,7 +46,6 @@ def __init__( self.groups = groups self.in_channels = in_channels self.out_channels = out_channels - self.activation = activation self.kernel_size = kernel_size self.small_kernel = small_kernel @@ -84,8 +81,6 @@ def forward(self, x: torch.Tensor) -> torch.Tensor: out = self.lkb_origin(x) if hasattr(self, "small_conv"): out += self.small_conv(x) - - self.activation(out) return out def get_kernel_bias(self) -> Tuple[torch.Tensor, torch.Tensor]: