anonauthors's picture
Add model
719a9f9 verified
{
"architecture": "swin_base_patch4_window7_224",
"num_classes": 10,
"num_features": 1024,
"global_pool": "avg",
"label_names": {
"airplane": 0,
"automobile": 1,
"bird": 2,
"cat": 3,
"deer": 4,
"dog": 5,
"frog": 6,
"horse": 7,
"ship": 8,
"truck": 9
},
"pretrained_cfg": {
"tag": "ms_in22k_ft_in1k",
"custom_load": false,
"input_size": [
3,
224,
224
],
"fixed_input_size": true,
"interpolation": "bicubic",
"crop_pct": 0.9,
"crop_mode": "center",
"mean": [
0.485,
0.456,
0.406
],
"std": [
0.229,
0.224,
0.225
],
"num_classes": 1000,
"pool_size": [
7,
7
],
"first_conv": "patch_embed.proj",
"classifier": "head.fc",
"license": "mit"
}
}