File tree Expand file tree Collapse file tree 1 file changed +5
-5
lines changed
tools/checkpoint_conversion Expand file tree Collapse file tree 1 file changed +5
-5
lines changed Original file line number Diff line number Diff line change 3
3
Optionally uploads the model to Keras if the `--upload_uri` flag is passed.
4
4
5
5
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
7
7
"""
8
8
9
9
import os
20
20
import keras_hub
21
21
22
22
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" ,
27
27
# TODO(jeffcarp): Add BN variants.
28
28
}
29
29
You can’t perform that action at this time.
0 commit comments