Merge pull request #1914 from developer0hye/developer0hye-patch-1

Set self.num_features to neck_chans if neck_chans > 0 for vision_transformer_sam

timm/models/vision_transformer_sam.py

@@ -434,6 +434,7 @@ class VisionTransformerSAM(nn.Module):
                 ),
                 LayerNorm2d(neck_chans),
             )
+            self.num_features = neck_chans
         else:
             self.neck = nn.Identity()
             neck_chans = embed_dim