Commit b219bbf4 authored by miguelCalado's avatar miguelCalado
Browse files

Added: Documentation and moved gpu.yaml file

parent c44de4b0
...@@ -152,6 +152,20 @@ python3 classifier_trainer.py \ ...@@ -152,6 +152,20 @@ python3 classifier_trainer.py \
--config_file=configs/examples/resnet/imagenet/tpu.yaml --config_file=configs/examples/resnet/imagenet/tpu.yaml
``` ```
### VGG-16
#### On GPU:
```bash
python3 classifier_trainer.py \
--mode=train_and_eval \
--model_type=vgg \
--dataset=imagenet \
--model_dir=$MODEL_DIR \
--data_dir=$DATA_DIR \
--config_file=configs/examples/vgg/imagenet/gpu.yaml \
--params_override='runtime.num_gpus=$NUM_GPUS'
```
### EfficientNet ### EfficientNet
**Note: EfficientNet development is a work in progress.** **Note: EfficientNet development is a work in progress.**
#### On GPU: #### On GPU:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment