Skip to content

Commit b219bbf

Browse files
author
miguelCalado
committed
Added: Documentation and moved gpu.yaml file
1 parent c44de4b commit b219bbf

File tree

2 files changed

+14
-0
lines changed

2 files changed

+14
-0
lines changed

official/legacy/image_classification/README.md

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -152,6 +152,20 @@ python3 classifier_trainer.py \
152152
--config_file=configs/examples/resnet/imagenet/tpu.yaml
153153
```
154154

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+
155169
### EfficientNet
156170
**Note: EfficientNet development is a work in progress.**
157171
#### On GPU:

0 commit comments

Comments
 (0)