Fix #2272

Author: rwightman

timm/models/vision_transformer.py

@@ -629,7 +629,7 @@ def reset_classifier(self, num_classes: int, global_pool: Optional[str] = None):
             assert global_pool in ('', 'avg', 'avgmax', 'max', 'token', 'map')
             if global_pool == 'map' and self.attn_pool is None:
                 assert False, "Cannot currently add attention pooling in reset_classifier()."
-            elif global_pool != 'map ' and self.attn_pool is not None:
+            elif global_pool != 'map' and self.attn_pool is not None:
                 self.attn_pool = None  # remove attention pooling
             self.global_pool = global_pool
         self.head = nn.Linear(self.embed_dim, num_classes) if num_classes > 0 else nn.Identity()