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
03038d9c
Commit
03038d9c
authored
Jan 22, 2020
by
Ayush Dubey
Committed by
A. Unique TensorFlower
Jan 22, 2020
Browse files
Update resnet README with multi-worker benchmark instructions.
PiperOrigin-RevId: 290963657
parent
d4eedbb9
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
13 additions
and
0 deletions
+13
-0
official/vision/image_classification/README.md
official/vision/image_classification/README.md
+13
-0
No files found.
official/vision/image_classification/README.md
View file @
03038d9c
...
@@ -113,6 +113,19 @@ distributed training across the GPUs.
...
@@ -113,6 +113,19 @@ distributed training across the GPUs.
If you wish to run without
`tf.distribute.Strategy`
, you can do so by setting
If you wish to run without
`tf.distribute.Strategy`
, you can do so by setting
`--distribution_strategy=off`
.
`--distribution_strategy=off`
.
### Running on multiple GPU hosts
You can also train these models on multiple hosts, each with GPUs, using
`tf.distribute.Strategy`
.
The easiest way to run multi-host benchmarks is to set the
[
`TF_CONFIG`
](
https://www.tensorflow.org/guide/distributed_training#TF_CONFIG
)
appropriately at each host. e.g., to run using
`MultiWorkerMirroredStrategy`
on
2 hosts, the
`cluster`
in
`TF_CONFIG`
should have 2
`host:port`
entries, and
host
`i`
should have the
`task`
in
`TF_CONFIG`
set to
`{"type": "worker",
"index": i}`
.
`MultiWorkerMirroredStrategy`
will automatically use all the
available GPUs at each host.
### Running on Cloud TPUs
### Running on Cloud TPUs
Note: This model will
**not**
work with TPUs on Colab.
Note: This model will
**not**
work with TPUs on Colab.
...
...
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