Skip to content

Commit a689a1c

Browse files
laxmareddyplaxmareddyp
authored
Update vit presets and Map Presets correct names in checkpoints (#2736)
1 parent 6c797e7 commit a689a1c

2 files changed

Lines changed: 22 additions & 22 deletions

File tree

‎keras_hub/src/models/vit/vit_presets.py‎

Lines changed: 11 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111
"params": 85798656,
1212
"path": "vit",
1313
},
14-
"kaggle_handle": "kaggle://keras/vit/keras/vit_base_patch16_224_imagenet/3",
14+
"kaggle_handle": "kaggle://keras/vit/keras/vit_base_patch16_224_imagenet/4",
1515
},
1616
"vit_base_patch16_384_imagenet": {
1717
"metadata": {
@@ -22,7 +22,7 @@
2222
"params": 86090496,
2323
"path": "vit",
2424
},
25-
"kaggle_handle": "kaggle://keras/vit/keras/vit_base_patch16_384_imagenet/3",
25+
"kaggle_handle": "kaggle://keras/vit/keras/vit_base_patch16_384_imagenet/4",
2626
},
2727
"vit_large_patch16_224_imagenet": {
2828
"metadata": {
@@ -33,7 +33,7 @@
3333
"params": 303301632,
3434
"path": "vit",
3535
},
36-
"kaggle_handle": "kaggle://keras/vit/keras/vit_large_patch16_224_imagenet/3",
36+
"kaggle_handle": "kaggle://keras/vit/keras/vit_large_patch16_224_imagenet/4",
3737
},
3838
"vit_large_patch16_384_imagenet": {
3939
"metadata": {
@@ -44,7 +44,7 @@
4444
"params": 303690752,
4545
"path": "vit",
4646
},
47-
"kaggle_handle": "kaggle://keras/vit/keras/vit_large_patch16_384_imagenet/3",
47+
"kaggle_handle": "kaggle://keras/vit/keras/vit_large_patch16_384_imagenet/4",
4848
},
4949
"vit_base_patch32_384_imagenet": {
5050
"metadata": {
@@ -55,7 +55,7 @@
5555
"params": 87528192,
5656
"path": "vit",
5757
},
58-
"kaggle_handle": "kaggle://keras/vit/keras/vit_base_patch32_384_imagenet/2",
58+
"kaggle_handle": "kaggle://keras/vit/keras/vit_base_patch32_384_imagenet/3",
5959
},
6060
"vit_large_patch32_384_imagenet": {
6161
"metadata": {
@@ -66,7 +66,7 @@
6666
"params": 305607680,
6767
"path": "vit",
6868
},
69-
"kaggle_handle": "kaggle://keras/vit/keras/vit_large_patch32_384_imagenet/2",
69+
"kaggle_handle": "kaggle://keras/vit/keras/vit_large_patch32_384_imagenet/3",
7070
},
7171
"vit_base_patch16_224_imagenet21k": {
7272
"metadata": {
@@ -77,7 +77,7 @@
7777
"params": 85798656,
7878
"path": "vit",
7979
},
80-
"kaggle_handle": "kaggle://keras/vit/keras/vit_base_patch16_224_imagenet21k/2",
80+
"kaggle_handle": "kaggle://keras/vit/keras/vit_base_patch16_224_imagenet21k/3",
8181
},
8282
"vit_base_patch32_224_imagenet21k": {
8383
"metadata": {
@@ -88,7 +88,7 @@
8888
"params": 87455232,
8989
"path": "vit",
9090
},
91-
"kaggle_handle": "kaggle://keras/vit/keras/vit_base_patch32_224_imagenet21k/2",
91+
"kaggle_handle": "kaggle://keras/vit/keras/vit_base_patch32_224_imagenet21k/3",
9292
},
9393
"vit_huge_patch14_224_imagenet21k": {
9494
"metadata": {
@@ -99,7 +99,7 @@
9999
"params": 630764800,
100100
"path": "vit",
101101
},
102-
"kaggle_handle": "kaggle://keras/vit/keras/vit_huge_patch14_224_imagenet21k/2",
102+
"kaggle_handle": "kaggle://keras/vit/keras/vit_huge_patch14_224_imagenet21k/3",
103103
},
104104
"vit_large_patch16_224_imagenet21k": {
105105
"metadata": {
@@ -110,7 +110,7 @@
110110
"params": 303301632,
111111
"path": "vit",
112112
},
113-
"kaggle_handle": "kaggle://keras/vit/keras/vit_large_patch16_224_imagenet21k/2",
113+
"kaggle_handle": "kaggle://keras/vit/keras/vit_large_patch16_224_imagenet21k/3",
114114
},
115115
"vit_large_patch32_224_imagenet21k": {
116116
"metadata": {
@@ -121,6 +121,6 @@
121121
"params": 305510400,
122122
"path": "vit",
123123
},
124-
"kaggle_handle": "kaggle://keras/vit/keras/vit_large_patch32_224_imagenet21k/2",
124+
"kaggle_handle": "kaggle://keras/vit/keras/vit_large_patch32_224_imagenet21k/3",
125125
},
126126
}

‎tools/checkpoint_conversion/convert_vit_checkpoints.py‎

Lines changed: 11 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -30,17 +30,17 @@
3030
FLAGS = flags.FLAGS
3131

3232
PRESET_MAP = {
33-
"vit_base_patch16_224": "google/vit-base-patch16-224",
34-
"vit_base_patch16_384": "google/vit-base-patch16-384",
35-
"vit_base_patch32_384": "google/vit-base-patch32-384",
36-
"vit_large_patch16_224": "google/vit-large-patch16-224",
37-
"vit_large_patch16_384": "google/vit-large-patch16-384",
38-
"vit_large_patch32_384": "google/vit-large-patch32-384",
39-
"vit_base_patch16_224_in21k": "google/vit-base-patch16-224-in21k",
40-
"vit_base_patch32_224_in21k": "google/vit-base-patch32-224-in21k",
41-
"vit_large_patch16_224_in21k": "google/vit-large-patch16-224-in21k",
42-
"vit_large_patch32_224_in21k": "google/vit-large-patch32-224-in21k",
43-
"vit_huge_patch14_224_in21k": "google/vit-huge-patch14-224-in21k",
33+
"vit_base_patch16_224_imagenet": "google/vit-base-patch16-224",
34+
"vit_base_patch16_384_imagenet": "google/vit-base-patch16-384",
35+
"vit_base_patch32_384_imagenet": "google/vit-base-patch32-384",
36+
"vit_large_patch16_224_imagenet": "google/vit-large-patch16-224",
37+
"vit_large_patch16_384_imagenet": "google/vit-large-patch16-384",
38+
"vit_large_patch32_384_imagenet": "google/vit-large-patch32-384",
39+
"vit_base_patch16_224_imagenet21k": "google/vit-base-patch16-224-in21k",
40+
"vit_base_patch32_224_imagenet21k": "google/vit-base-patch32-224-in21k",
41+
"vit_large_patch16_224_imagenet21k": "google/vit-large-patch16-224-in21k",
42+
"vit_large_patch32_224_imagenet21k": "google/vit-large-patch32-224-in21k",
43+
"vit_huge_patch14_224_imagenet21k": "google/vit-huge-patch14-224-in21k",
4444
}
4545

4646
flags.DEFINE_string(

0 commit comments

Comments
 (0)