image size option

2023-10-15 23:48:14 +01:00
parent cf9c34c801
commit 62746e7f03
3 changed files with 6 additions and 3 deletions
@@ -49,7 +49,7 @@ class VisionTransformer(timm.models.vision_transformer.VisionTransformer):

 def vit_large_patch16(**kwargs):
    model = VisionTransformer(
-        img_size=224,patch_size=16, embed_dim=1024, depth=24, num_heads=16, mlp_ratio=4, qkv_bias=True,
+        patch_size=16, embed_dim=1024, depth=24, num_heads=16, mlp_ratio=4, qkv_bias=True,
        norm_layer=partial(nn.LayerNorm, eps=1e-6), **kwargs)
    return model