From c33a0013977c006f993d55c2c6fbc60e2c989093 Mon Sep 17 00:00:00 2001 From: Sejik <52148895+Sejik@users.noreply.github.com> Date: Mon, 24 Jun 2024 21:54:38 +0900 Subject: [PATCH] Fix typo --- timm/models/vision_transformer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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.