diff --git a/timm/models/vision_transformer.py b/timm/models/vision_transformer.py index 441ac0c5..9a3ac562 100644 --- a/timm/models/vision_transformer.py +++ b/timm/models/vision_transformer.py @@ -460,7 +460,7 @@ class VisionTransformer(nn.Module): img_size: Input image size. patch_size: Patch size. in_chans: Number of image input channels. - num_classes: Mumber of classes for classification head. + num_classes: Number of classes for classification head. global_pool: Type of global pooling for final sequence (default: 'token'). embed_dim: Transformer embedding dimension. depth: Depth of transformer.