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
33298ffd
Commit
33298ffd
authored
Aug 24, 2017
by
Toby Boyd
Browse files
Minor edits to generate tfrecords and headers
parent
b9c582cc
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
15 additions
and
16 deletions
+15
-16
tutorials/image/cifar10_estimator/README.md
tutorials/image/cifar10_estimator/README.md
+15
-16
No files found.
tutorials/image/cifar10_estimator/README.md
View file @
33298ffd
...
@@ -4,6 +4,7 @@ http://www.cs.toronto.edu/~kriz/cifar.html
...
@@ -4,6 +4,7 @@ http://www.cs.toronto.edu/~kriz/cifar.html
Code in this directory focuses on how to use TensorFlow Estimators to train and
Code in this directory focuses on how to use TensorFlow Estimators to train and
evaluate a CIFAR-10 ResNet model on:
evaluate a CIFAR-10 ResNet model on:
*
A single host with one CPU;
*
A single host with one CPU;
*
A single host with multiple GPUs;
*
A single host with multiple GPUs;
*
Multiple hosts with CPU or multiple GPUs;
*
Multiple hosts with CPU or multiple GPUs;
...
@@ -12,31 +13,29 @@ Before trying to run the model we highly encourage you to read all the README.
...
@@ -12,31 +13,29 @@ Before trying to run the model we highly encourage you to read all the README.
## Prerequisite
## Prerequisite
1.
Install TensorFlow version 1.2.1 or
later with GPU support.
1.
[
Install
](
https://www.tensorflow.org/install/
)
TensorFlow version 1.2.1 or
You can see how to do it
[
here
](
https://www.tensorflow.org/install/
)
.
later
.
2.
Generate TFRecord files.
2.
Download the CIFAR-10 dataset and generate TFRecord files using the provided
This will generate a tf record for the training and test data available at the
script. The script and associated command below will download the CIFAR-10
input_dir. You can see more details in
`generate_cifar10_tf_records.py`
dataset and then generate a TFRecord for the training, validation, and
evaluation datasets.
```
shell
```
shell
python generate_cifar10_tfrecords.py
--data-dir
=
${
PWD
}
/cifar-10-data
python generate_cifar10_tfrecords.py
--data-dir
=
${
PWD
}
/cifar-10-data
```
```
After running the command above, you should see the following
new
files in the
After running the command above, you should see the following files in the
output_dir.
--data-dir (
```ls -R cifar-10-data```
):
```
shell
*
train.tfrecords
ls
-R
cifar-10-data
*
validation.tfrecords
```
*
eval.tfrecords
```
train.tfrecords validation.tfrecords eval.tfrecords
```
##
How to run on local mode
##
Training on a single machine with GPUs or CPU
Run the
model
on CPU only. After training, it runs the evaluation.
Run the
training
on CPU only. After training, it runs the evaluation.
```
```
python cifar10_main.py --data-dir=${PWD}/cifar-10-data \
python cifar10_main.py --data-dir=${PWD}/cifar-10-data \
...
@@ -69,7 +68,7 @@ python cifar10_main.py --data-dir=${PWD}/cifar-10-data \
...
@@ -69,7 +68,7 @@ python cifar10_main.py --data-dir=${PWD}/cifar-10-data \
There are more command line flags to play with; run
There are more command line flags to play with; run
`python cifar10_main.py --help`
for details.
`python cifar10_main.py --help`
for details.
##
How to run o
n distributed
mode
##
Ru
n distributed
training
### (Optional) Running on Google Cloud Machine Learning Engine
### (Optional) Running on Google Cloud Machine Learning Engine
...
...
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