diff --git a/timm/models/sequencer.py b/timm/models/sequencer.py index 88003540..3ffaf02b 100644 --- a/timm/models/sequencer.py +++ b/timm/models/sequencer.py @@ -276,7 +276,7 @@ class Sequencer2D(nn.Module): ): super().__init__() self.num_classes = num_classes - self.num_features = embed_dims[0] # num_features for consistency with other models + self.num_features = embed_dims[-1] # num_features for consistency with other models self.embed_dims = embed_dims self.stem = PatchEmbed( img_size=img_size, patch_size=patch_sizes[0], in_chans=in_chans,