Update model config
Browse files- config.json +2 -0
config.json
CHANGED
@@ -4,6 +4,7 @@
|
|
4 |
"num_features": 512,
|
5 |
"global_pool": "avg",
|
6 |
"pretrained_cfg": {
|
|
|
7 |
"custom_load": false,
|
8 |
"input_size": [
|
9 |
3,
|
@@ -25,6 +26,7 @@
|
|
25 |
0.5
|
26 |
],
|
27 |
"num_classes": 11821,
|
|
|
28 |
"first_conv": "patch_embed.proj",
|
29 |
"classifier": "head"
|
30 |
}
|
|
|
4 |
"num_features": 512,
|
5 |
"global_pool": "avg",
|
6 |
"pretrained_cfg": {
|
7 |
+
"tag": "in12k",
|
8 |
"custom_load": false,
|
9 |
"input_size": [
|
10 |
3,
|
|
|
26 |
0.5
|
27 |
],
|
28 |
"num_classes": 11821,
|
29 |
+
"pool_size": null,
|
30 |
"first_conv": "patch_embed.proj",
|
31 |
"classifier": "head"
|
32 |
}
|