"test/vscode:/vscode.git/clone" did not exist on "d27a6f7092cf7caa5b99089d78c692a529ff8dd2"
Commit a5f64716 authored by Marianne Linhares Monteiro's avatar Marianne Linhares Monteiro Committed by GitHub
Browse files

Small fixes in README

- 2 CPUs -> 2 GPUs
- is_cpu_ps=True
parent 05efe3aa
......@@ -33,10 +33,10 @@ $ python cifar10_main.py --data_dir=/prefix/to/downloaded/data/cifar-10-batches-
--num_gpus=0 \
--train_steps=1000
# Run the model on CPU and 2 CPUs. After training, it runs the evaluation.
# Run the model on CPU and 2 GPUs. 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=False \
--is_cpu_ps=True \
--force_gpu_compatible=True \
--num_gpus=2 \
--train_steps=1000
......
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