{ | |
"image_size": 448, | |
"model_name": "vit_large", | |
"model_args": { | |
"patch_size": 14, | |
"num_classes": 10861, | |
"num_layers": 24, | |
"embed_dim": 1024, | |
"mlp_dim": 4096, | |
"num_heads": 16, | |
"drop_path_rate": 0.1, | |
"norm_layer": "reparam_layernorm", | |
"layer_norm_eps": 1e-05 | |
} | |
} |