We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent c44de4b commit b219bbfCopy full SHA for b219bbf
official/legacy/image_classification/README.md
@@ -152,6 +152,20 @@ python3 classifier_trainer.py \
152
--config_file=configs/examples/resnet/imagenet/tpu.yaml
153
```
154
155
+### VGG-16
156
+
157
+#### On GPU:
158
+```bash
159
+python3 classifier_trainer.py \
160
+ --mode=train_and_eval \
161
+ --model_type=vgg \
162
+ --dataset=imagenet \
163
+ --model_dir=$MODEL_DIR \
164
+ --data_dir=$DATA_DIR \
165
+ --config_file=configs/examples/vgg/imagenet/gpu.yaml \
166
+ --params_override='runtime.num_gpus=$NUM_GPUS'
167
+```
168
169
### EfficientNet
170
**Note: EfficientNet development is a work in progress.**
171
#### On GPU:
official/legacy/image_classification/configs/examples/vgg16/gpu.yaml renamed to official/legacy/image_classification/configs/examples/vgg16/imagenet/gpu.yaml
0 commit comments