From 93a79a3dd99f53e759525a5be945e8ba93678009 Mon Sep 17 00:00:00 2001 From: okojoalg Date: Fri, 6 May 2022 23:16:32 +0900 Subject: [PATCH] Fix num_features in Sequencer --- timm/models/sequencer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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,