vitmodel4 / config.json
tmshag1's picture
Add model
1a3f9f6
raw
history blame contribute delete
558 Bytes
{
"url": "https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-tresnet/vit_base_patch16_224_1k_miil_84_4-2deb18e3.pth",
"num_classes": 1,
"input_size": [
3,
224,
224
],
"pool_size": null,
"crop_pct": 0.875,
"interpolation": "bilinear",
"fixed_input_size": true,
"mean": [
0.0,
0.0,
0.0
],
"std": [
1.0,
1.0,
1.0
],
"first_conv": "patch_embed.proj",
"classifier": "head",
"architecture": "vit_base_patch16_224_miil",
"num_features": 768,
"labels": [
"LABEL_0"
]
}