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
a6f10ec0
Commit
a6f10ec0
authored
Aug 20, 2019
by
Haoyu Zhang
Committed by
A. Unique TensorFlower
Aug 20, 2019
Browse files
Fix filenames and links in Keras ResNet README.
PiperOrigin-RevId: 264411162
parent
3858c82b
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
14 additions
and
14 deletions
+14
-14
official/vision/image_classification/README.md
official/vision/image_classification/README.md
+14
-14
No files found.
official/vision/image_classification/README.md
View file @
a6f10ec0
...
@@ -18,20 +18,20 @@ official.resnet`.
...
@@ -18,20 +18,20 @@ official.resnet`.
Download and extract the CIFAR-10 data. You can use the following script:
Download and extract the CIFAR-10 data. You can use the following script:
```
bash
```
bash
python cifar10_download_and_extract.py
python
../../r1/resnet/
cifar10_download_and_extract.py
```
```
After you download the data, you can run the program by:
After you download the data, you can run the program by:
```
bash
```
bash
python
keras
_cifar_main.py
python
resnet
_cifar_main.py
```
```
If you did not use the default directory to download the data, specify the
If you did not use the default directory to download the data, specify the
location with the
`--data_dir`
flag, like:
location with the
`--data_dir`
flag, like:
```
bash
```
bash
python
keras
_cifar_main.py
--data_dir
=
/path/to/cifar
python
resnet
_cifar_main.py
--data_dir
=
/path/to/cifar
```
```
## ImageNet
## ImageNet
...
@@ -44,14 +44,14 @@ provide a few options.
...
@@ -44,14 +44,14 @@ provide a few options.
Once your dataset is ready, you can begin training the model as follows:
Once your dataset is ready, you can begin training the model as follows:
```
bash
```
bash
python
keras
_imagenet_main.py
python
resnet
_imagenet_main.py
```
```
Again, if you did not download the data to the default directory, specify the
Again, if you did not download the data to the default directory, specify the
location with the
`--data_dir`
flag:
location with the
`--data_dir`
flag:
```
bash
```
bash
python
keras
_imagenet_main.py
--data_dir
=
/path/to/imagenet
python
resnet
_imagenet_main.py
--data_dir
=
/path/to/imagenet
```
```
There are more flag options you can specify. Here are some examples:
There are more flag options you can specify. Here are some examples:
...
@@ -70,16 +70,16 @@ For example, this is a typical command line to run with ImageNet data with
...
@@ -70,16 +70,16 @@ For example, this is a typical command line to run with ImageNet data with
batch size 128 per GPU:
batch size 128 per GPU:
```
bash
```
bash
python
-m
keras
_imagenet_main
\
python
-m
resnet
_imagenet_main
\
--model_dir
=
/tmp/model_dir/something
\
--model_dir
=
/tmp/model_dir/something
\
--num_gpus
=
2
\
--num_gpus
=
2
\
--batch_size
=
128
\
--batch_size
=
128
\
--train_epochs
=
90
\
--train_epochs
=
90
\
--train_steps
=
10
\
--train_steps
=
10
\
--use_synthetic_data
=
false
--use_synthetic_data
=
false
```
```
See
[
`
keras_
common.py`
](
keras_
common.py
)
for full list of options.
See
[
`common.py`
](
common.py
)
for full list of options.
## Using multiple GPUs
## Using multiple GPUs
You can train these models on multiple GPUs using
`tf.distribute.Strategy`
API.
You can train these models on multiple GPUs using
`tf.distribute.Strategy`
API.
...
...
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