Image Feature Extraction
Birder
PyTorch
vit_l16_mim / vit_l16_mim.json
hassonofer's picture
Update vit_l16_mim.json
37a242d verified
{
"birder_version": "v0.0.5a91",
"name": "vit",
"alias": "vit_l16",
"task": "image_classification",
"net_param": null,
"model_config": {
"patch_size": 16,
"num_layers": 24,
"num_heads": 16,
"hidden_dim": 1024,
"mlp_dim": 4096,
"num_reg_tokens": 0,
"drop_path_rate": 0.1
},
"signature": {
"inputs": [
{
"data_shape": [
0,
3,
224,
224
]
}
],
"outputs": [
{
"data_shape": [
0,
0
]
}
]
},
"rgb_stats": {
"mean": [
0.5248,
0.5372,
0.5086
],
"std": [
0.2135,
0.2103,
0.2622
]
}
}