Commit 30268a43 authored by Toby Boyd's avatar Toby Boyd Committed by GitHub
Browse files

Merge pull request #1945 from mari-linhares/patch-1

Small fixes in README
parents 1f6bfcfa 61065b9d
...@@ -33,7 +33,15 @@ $ python cifar10_main.py --data_dir=/prefix/to/downloaded/data/cifar-10-batches- ...@@ -33,7 +33,15 @@ $ python cifar10_main.py --data_dir=/prefix/to/downloaded/data/cifar-10-batches-
--num_gpus=0 \ --num_gpus=0 \
--train_steps=1000 --train_steps=1000
# Run the model on CPU and 2 CPUs. After training, it runs the evaluation. # Run the model on 2 GPUs using CPU as parameter server. After training, it runs the evaluation.
$ python cifar10_main.py --data_dir=/prefix/to/downloaded/data/cifar-10-batches-bin \
--model_dir=/tmp/cifar10 \
--is_cpu_ps=True \
--force_gpu_compatible=True \
--num_gpus=2 \
--train_steps=1000
# Run the model on 2 GPUs using GPU as parameter server. After training, it runs the evaluation.
$ python cifar10_main.py --data_dir=/prefix/to/downloaded/data/cifar-10-batches-bin \ $ python cifar10_main.py --data_dir=/prefix/to/downloaded/data/cifar-10-batches-bin \
--model_dir=/tmp/cifar10 \ --model_dir=/tmp/cifar10 \
--is_cpu_ps=False \ --is_cpu_ps=False \
......
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