Fix num_features in Sequencer

2 years ago · 93a79a3dd9
parent 2fec08e923
commit 93a79a3dd9
1 changed files with 1 additions and 1 deletions
--- 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,