Commit dd889bf4 authored by James Qin's avatar James Qin
Browse files

Revise README

parent 47b0355e
...@@ -2,13 +2,13 @@ CIFAR-10 is a common benchmark in machine learning for image recognition. ...@@ -2,13 +2,13 @@ CIFAR-10 is a common benchmark in machine learning for image recognition.
http://www.cs.toronto.edu/~kriz/cifar.html http://www.cs.toronto.edu/~kriz/cifar.html
Code in this directory focus on how to use TensorFlow Estimator to train and evaluate CIFAR-10ResetNet model, on a single host with one CPU and potentially multiple GPUs. Code in this directory focus on how to use TensorFlow Estimator to train and evaluate CIFAR-10 ResetNet model, on a single host with one CPU and potentially multiple GPUs.
<b>Prerequisite:</b> <b>Prerequisite:</b>
1. Install TensorFlow with GPU support with a version more recent than 1.2.0-cp27. 1. Install TensorFlow with GPU support with a version more recent than 1.2.0-cp27.
2. Download CIFAR-10 dataset. 2. Download the CIFAR-10 dataset.
```shell ```shell
curl -o cifar-10-python.tar.gz https://www.cs.toronto.edu/~kriz/cifar-10-python.tar.gz curl -o cifar-10-python.tar.gz https://www.cs.toronto.edu/~kriz/cifar-10-python.tar.gz
...@@ -41,5 +41,5 @@ $ python cifar10_main.py --data_dir=/prefix/to/downloaded/data/cifar-10-batches- ...@@ -41,5 +41,5 @@ $ python cifar10_main.py --data_dir=/prefix/to/downloaded/data/cifar-10-batches-
--num_gpus=2 \ --num_gpus=2 \
--train_steps=1000 --train_steps=1000
# There are more commandline flags to play with, check cifar10_main.py for details. # There are more command line flags to play with; check cifar10_main.py for details.
``` ```
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