Skip to content

Commit 6df4820

Browse files
Update preset map for VGG model (#2411)
* Update preset map for VGG model * Update preset map for VGG model
1 parent b51b273 commit 6df4820

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

tools/checkpoint_conversion/convert_vgg_checkpoints.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
Optionally uploads the model to Keras if the `--upload_uri` flag is passed.
44
55
python tools/checkpoint_conversion/convert_vgg_checkpoints.py \
6-
--preset vgg11 --upload_uri kaggle://kerashub/vgg/keras/vgg11
6+
--preset vgg_11_imagenet --upload_uri kaggle://kerashub/vgg/keras/vgg_11_imagenet
77
"""
88

99
import os
@@ -20,10 +20,10 @@
2020
import keras_hub
2121

2222
PRESET_MAP = {
23-
"vgg11": "timm/vgg11.tv_in1k",
24-
"vgg13": "timm/vgg13.tv_in1k",
25-
"vgg16": "timm/vgg16.tv_in1k",
26-
"vgg19": "timm/vgg19.tv_in1k",
23+
"vgg_11_imagenet": "timm/vgg11.tv_in1k",
24+
"vgg_13_imagenet": "timm/vgg13.tv_in1k",
25+
"vgg_16_imagenet": "timm/vgg16.tv_in1k",
26+
"vgg_19_imagenet": "timm/vgg19.tv_in1k",
2727
# TODO(jeffcarp): Add BN variants.
2828
}
2929

0 commit comments

Comments
 (0)