Skip to content
Snippets Groups Projects
Commit e530cc34 authored by gent's avatar gent
Browse files

temp p for tiny

parent 41776001
No related branches found
No related tags found
No related merge requests found
......@@ -19,7 +19,7 @@ class VisionTransformer(timm.models.vision_transformer.VisionTransformer):
def vit_tiny(**kwargs):
model = VisionTransformer(
patch_size=16,embed_dim=192,depth=12,num_heads=12,, mlp_ratio=4, qkv_bias=True,
patch_size=16,embed_dim=192,depth=12,num_heads=12, mlp_ratio=4, qkv_bias=True,
norm_layer=partial(nn.LayerNorm, eps=1e-6), **kwargs)
return model
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment