mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

flip these two

Browse Source
This commit is contained in:
Yassine 2023-09-26 19:40:46 -07:00 committed by Ross Wightman
parent 8ba2038e6b
commit 5c504b4ded
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
timm/models/fastvit.py
View File

@ -421,8 +421,8 @@ class ReparamLargeKernelConv(nn.Module):
def convolutional_stem( def convolutional_stem(
in_chs: int, in_chs: int,
out_chs: int, out_chs: int,
inference_mode: bool = False,
act_layer: nn.Module = nn.GELU, act_layer: nn.Module = nn.GELU,
inference_mode: bool = False
) -> nn.Sequential: ) -> nn.Sequential:
"""Build convolutional stem with MobileOne blocks. """Build convolutional stem with MobileOne blocks.
@ -1124,8 +1124,8 @@ class FastVit(nn.Module):
self.stem = convolutional_stem( self.stem = convolutional_stem(
in_chans, in_chans,
embed_dims[0], embed_dims[0],
act_layer,
inference_mode, inference_mode,
act_layer
) )
# Build the main stages of the network architecture # Build the main stages of the network architecture
@ -1187,6 +1187,7 @@ class FastVit(nn.Module):
group_size=1, group_size=1,
inference_mode=inference_mode, inference_mode=inference_mode,
use_se=True, use_se=True,
act_layer=act_layer,
num_conv_branches=1, num_conv_branches=1,
) )
self.head = ClassifierHead( self.head = ClassifierHead(