From 5c504b4ded921565b8e8d22488cac14754b2956a Mon Sep 17 00:00:00 2001 From: Yassine Date: Tue, 26 Sep 2023 19:40:46 -0700 Subject: [PATCH] flip these two --- timm/models/fastvit.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/timm/models/fastvit.py b/timm/models/fastvit.py index fbeeb39e..d3d9bfdf 100644 --- a/timm/models/fastvit.py +++ b/timm/models/fastvit.py @@ -421,8 +421,8 @@ class ReparamLargeKernelConv(nn.Module): def convolutional_stem( in_chs: int, out_chs: int, - inference_mode: bool = False, act_layer: nn.Module = nn.GELU, + inference_mode: bool = False ) -> nn.Sequential: """Build convolutional stem with MobileOne blocks. @@ -1124,8 +1124,8 @@ class FastVit(nn.Module): self.stem = convolutional_stem( in_chans, embed_dims[0], + act_layer, inference_mode, - act_layer ) # Build the main stages of the network architecture @@ -1187,6 +1187,7 @@ class FastVit(nn.Module): group_size=1, inference_mode=inference_mode, use_se=True, + act_layer=act_layer, num_conv_branches=1, ) self.head = ClassifierHead(