Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
ResNet50_tensorflow
Commits
51d99c66
Commit
51d99c66
authored
Jul 16, 2017
by
Marianne Linhares Monteiro
Committed by
GitHub
Jul 16, 2017
Browse files
Adding another example for is_cpu_ps=False
parent
a5f64716
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
9 additions
and
1 deletion
+9
-1
tutorials/image/cifar10_estimator/README.md
tutorials/image/cifar10_estimator/README.md
+9
-1
No files found.
tutorials/image/cifar10_estimator/README.md
View file @
51d99c66
...
...
@@ -33,7 +33,7 @@ $ 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 GPUs
. 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
\
...
...
@@ -41,5 +41,13 @@ $ python cifar10_main.py --data_dir=/prefix/to/downloaded/data/cifar-10-batches-
--num_gpus
=
2
\
--train_steps
=
1000
# Run the model on 2 GPUs usinc GPU 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
=
False
\
--force_gpu_compatible
=
True
\
--num_gpus
=
2
\
--train_steps
=
1000
# There are more command line flags to play with; check cifar10_main.py for details.
```
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment