Unverified Commit 4698a41e authored by Toby Boyd's avatar Toby Boyd Committed by GitHub
Browse files

Small word tweak (#6650)

* Small word tweak

* Few more tweaks
parent c96422f7
...@@ -35,16 +35,16 @@ Please proceed according to which dataset you would like to train/evaluate on: ...@@ -35,16 +35,16 @@ Please proceed according to which dataset you would like to train/evaluate on:
### Setup ### Setup
You simply need to have the latest version of TensorFlow installed. You need to have the latest version of TensorFlow installed.
First make sure you've [added the models folder to your Python path](/official/#running-the-models); otherwise you may encounter an error like `ImportError: No module named official.resnet`. First, make sure [the models folder is in your Python path](/official/#running-the-models); otherwise you may encounter `ImportError: No module named official.resnet`.
Then download and extract the CIFAR-10 data from Alex's website, specifying the location with the `--data_dir` flag. Run the following: Then, download and extract the CIFAR-10 data from Alex's website, specifying the location with the `--data_dir` flag. Run the following:
```bash ```bash
python cifar10_download_and_extract.py --data_dir <DATA_DIR> python cifar10_download_and_extract.py --data_dir <DATA_DIR>
``` ```
Then to train the model, run the following: Then, to train the model:
```bash ```bash
python cifar10_main.py --data_dir <DATA_DIR>/cifar-10-batches-bin --model_dir <MODEL_DIR> python cifar10_main.py --data_dir <DATA_DIR>/cifar-10-batches-bin --model_dir <MODEL_DIR>
...@@ -52,13 +52,13 @@ python cifar10_main.py --data_dir <DATA_DIR>/cifar-10-batches-bin --model_dir <M ...@@ -52,13 +52,13 @@ python cifar10_main.py --data_dir <DATA_DIR>/cifar-10-batches-bin --model_dir <M
Use `--data_dir` to specify the location of the CIFAR-10 data used in the previous step. There are more flag options as described in `cifar10_main.py`. Use `--data_dir` to specify the location of the CIFAR-10 data used in the previous step. There are more flag options as described in `cifar10_main.py`.
To export a `SavedModel` from the trained checkpoint, run the following: To export a `SavedModel` from the trained checkpoint:
```bash ```bash
python cifar10_main.py --data_dir <DATA_DIR>/cifar-10-batches-bin --model_dir <MODEL_DIR> --eval_only --export_dir <EXPORT_DIR> python cifar10_main.py --data_dir <DATA_DIR>/cifar-10-batches-bin --model_dir <MODEL_DIR> --eval_only --export_dir <EXPORT_DIR>
``` ```
Note the `<EXPORT_DIR>` must be present. You might want to run `mkdir <EXPORT_DIR>` beforehand. Note: The `<EXPORT_DIR>` must be present. You might want to run `mkdir <EXPORT_DIR>` beforehand.
The `SavedModel` can then be [loaded](https://www.tensorflow.org/guide/saved_model#loading_a_savedmodel_in_python) in order to use the ResNet for prediction. The `SavedModel` can then be [loaded](https://www.tensorflow.org/guide/saved_model#loading_a_savedmodel_in_python) in order to use the ResNet for prediction.
......
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