File size: 1,392 Bytes
bc443ab | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 | {
"model_id_list": [
"vit_base_patch8_224"
"vit_base_patch16_224.augreg2_in21k_ft_in1k", # with finetuned head
"vit_base_patch16_224", # backbone only
"vit_large_patch16_rope_224",
"vit_huge_patch16_224",
"vit_large_patch14_224", # backbone only
"vit_huge_patch14_224", # backbone only
# DeiT3 ViT Models -img resolution 224 (Have finetuned head)
"deit3_small_patch16_224.fb_in22k_ft_in1k",
"deit3_base_patch16_224.fb_in22k_ft_in1k",
"deit3_medium_patch16_224.fb_in22k_ft_in1k",
"deit3_large_patch16_224.fb_in22k_ft_in1k",
# DeiT3 ViT Models -img resolution 384
"deit3_small_patch16_384.fb_in22k_ft_in1k",
"deit3_base_patch16_384.fb_in22k_ft_in1k",
"deit3_large_patch16_384.fb_in22k_ft_in1k",
# DinoV2
"vit_small_patch14_dinov2",
"vit_base_patch14_dinov2" # I finetuned this head; model checkpoint - vit_base_patch14_dinov2_in1k_best
# DinoV2 ViT Models -img resolution 518 with registers
"vit_small_patch14_reg4_dinov2",
"vit_base_patch14_reg4_dinov2", # I finetuned this head
# DinoV3
"vit_base_patch16_dinov3_in1k",
"vit_base_patch16_dinov3",# I finetuned this head # TODO: what is the difference between this and the one above?
"vit_base_patch16_dinov3_qkvb", #
""
]
}
|