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
d067ce0a
Commit
d067ce0a
authored
Aug 16, 2017
by
Eli Bixby
Browse files
Fix readme + argparse
parent
30fa4ebb
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
4 additions
and
14 deletions
+4
-14
tutorials/image/cifar10_estimator/README.md
tutorials/image/cifar10_estimator/README.md
+4
-8
tutorials/image/cifar10_estimator/cifar10_main.py
tutorials/image/cifar10_estimator/cifar10_main.py
+0
-6
No files found.
tutorials/image/cifar10_estimator/README.md
View file @
d067ce0a
...
@@ -73,8 +73,7 @@ $ python cifar10_main.py --data-dir=/prefix/to/downloaded/data/cifar-10-batches-
...
@@ -73,8 +73,7 @@ $ python cifar10_main.py --data-dir=/prefix/to/downloaded/data/cifar-10-batches-
--is-cpu-ps=False \
--is-cpu-ps=False \
--force-gpu-compatible=True \
--force-gpu-compatible=True \
--num-gpus=2 \
--num-gpus=2 \
--train-steps=1000
# There are more command line 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.
```
```
...
@@ -99,19 +98,16 @@ Then run the following command from the `tutorials/image` directory of this repo
...
@@ -99,19 +98,16 @@ Then run the following command from the `tutorials/image` directory of this repo
```
```
gcloud ml-engine jobs submit training cifarmultigpu \
gcloud ml-engine jobs submit training cifarmultigpu \
--runtime-version 1.2 \
--runtime-version 1.2 \
--
staging-bucket $MY_BUCKET
\
--
job-dir=$MY_BUCKET/model_dirs/cifarmultigpu
\
--config cifar10_estimator/
job
_config.yaml \
--config cifar10_estimator/
cmle
_config.yaml \
--package-path cifar10_estimator/ \
--package-path cifar10_estimator/ \
--region us-central1 \
--module-name cifar10_estimator.cifar10_main \
--module-name cifar10_estimator.cifar10_main \
-- \
-- \
--data-dir=$MY_BUCKET/cifar-10-batches-py \
--data-dir=$MY_BUCKET/cifar-10-batches-py \
--job-dir=$MY_BUCKLET/model_dirs/cifarmultigpu \
--is-cpu-ps=True \
--is-cpu-ps=True \
--force-gpu-compatible=True \
--force-gpu-compatible=True \
--num-gpus=4 \
--num-gpus=4 \
--train-steps=1000 \
--train-steps=1000
```
```
...
...
tutorials/image/cifar10_estimator/cifar10_main.py
View file @
d067ce0a
...
@@ -559,12 +559,6 @@ if __name__ == '__main__':
...
@@ -559,12 +559,6 @@ if __name__ == '__main__':
If present when running in a distributed environment will run on sync mode.
\
If present when running in a distributed environment will run on sync mode.
\
"""
"""
)
)
parser
.
add_argument
(
'--num-workers'
,
type
=
int
,
default
=
1
,
help
=
'Number of workers.'
)
parser
.
add_argument
(
parser
.
add_argument
(
'--num-intra-threads'
,
'--num-intra-threads'
,
type
=
int
,
type
=
int
,
...
...
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