rwood-97's picture
Add model
3c996fc
raw
history blame
618 Bytes
{
"url": "https://storage.googleapis.com/vit_models/augreg/B_16-i21k-300ep-lr_0.001-aug_medium1-wd_0.1-do_0.0-sd_0.0--imagenet2012-steps_20k-lr_0.01-res_224.npz",
"num_classes": 4,
"input_size": [
3,
224,
224
],
"pool_size": null,
"crop_pct": 0.9,
"interpolation": "bicubic",
"fixed_input_size": true,
"mean": [
0.5,
0.5,
0.5
],
"std": [
0.5,
0.5,
0.5
],
"first_conv": "patch_embed.proj",
"classifier": "head",
"architecture": "vit_base_patch16_224",
"num_features": 768,
"labels": [
"LABEL_0",
"LABEL_1",
"LABEL_2",
"LABEL_3"
]
}