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
aa596195
Commit
aa596195
authored
Mar 30, 2022
by
A. Unique TensorFlower
Browse files
Correcting some typos, formatting etc.
PiperOrigin-RevId: 438363336
parent
01824447
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
31 additions
and
9 deletions
+31
-9
official/legacy/image_classification/README.md
official/legacy/image_classification/README.md
+29
-6
official/vision/beta/README.md
official/vision/beta/README.md
+2
-3
No files found.
official/legacy/image_classification/README.md
View file @
aa596195
# Image Classification
# Image Classification
**Warning:**
the features in the
`image_classification/`
folder
have been
fully
**Warning:**
the features in the
`image_classification/`
directory
have been
integrated into
vision/beta. Please use the
[
new code base
](
../../vision/beta/README.md
)
.
fully
integrated into
the
[
new code base
](
https://github.com/tensorflow/models/tree/benchmark/official/vision/modeling/backbones
)
.
This folder contains TF 2
.0
model examples for image classification:
This folder contains TF 2 model examples for image classification:
*
[
MNIST
](
#mnist
)
*
[
MNIST
](
#mnist
)
*
[
Classifier Trainer
](
#classifier-trainer
)
, a framework that uses the Keras
*
[
Classifier Trainer
](
#classifier-trainer
)
, a framework that uses the Keras
...
@@ -17,8 +17,7 @@ For more information about other types of models, please refer to this
...
@@ -17,8 +17,7 @@ For more information about other types of models, please refer to this
## Before you begin
## Before you begin
Please make sure that you have the latest version of TensorFlow
Please make sure that you have the latest version of TensorFlow
installed and
installed and add the models folder to your Python path.
[
add the models folder to your Python path
](
/official/#running-the-models
)
.
### ImageNet preparation
### ImageNet preparation
...
@@ -70,6 +69,7 @@ available GPUs at each host.
...
@@ -70,6 +69,7 @@ available GPUs at each host.
To download the data and run the MNIST sample model locally for the first time,
To download the data and run the MNIST sample model locally for the first time,
run one of the following command:
run one of the following command:
<details>
```
bash
```
bash
python3 mnist_main.py
\
python3 mnist_main.py
\
--model_dir
=
$MODEL_DIR
\
--model_dir
=
$MODEL_DIR
\
...
@@ -79,9 +79,11 @@ python3 mnist_main.py \
...
@@ -79,9 +79,11 @@ python3 mnist_main.py \
--num_gpus
=
$NUM_GPUS
\
--num_gpus
=
$NUM_GPUS
\
--download
--download
```
```
</details>
To train the model on a Cloud TPU, run the following command:
To train the model on a Cloud TPU, run the following command:
<details>
```
bash
```
bash
python3 mnist_main.py
\
python3 mnist_main.py
\
--tpu
=
$TPU_NAME
\
--tpu
=
$TPU_NAME
\
...
@@ -91,10 +93,10 @@ python3 mnist_main.py \
...
@@ -91,10 +93,10 @@ python3 mnist_main.py \
--distribution_strategy
=
tpu
\
--distribution_strategy
=
tpu
\
--download
--download
```
```
</details>
Note: the
`--download`
flag is only required the first time you run the model.
Note: the
`--download`
flag is only required the first time you run the model.
## Classifier Trainer
## Classifier Trainer
The classifier trainer is a unified framework for running image classification
The classifier trainer is a unified framework for running image classification
models using Keras's compile/fit methods. Experiments should be provided in the
models using Keras's compile/fit methods. Experiments should be provided in the
...
@@ -111,6 +113,8 @@ be 64 * 8 = 512, and for a v3-32, the global batch size is 64 * 32 = 2048.
...
@@ -111,6 +113,8 @@ be 64 * 8 = 512, and for a v3-32, the global batch size is 64 * 32 = 2048.
### ResNet50
### ResNet50
#### On GPU:
#### On GPU:
<details>
```
bash
```
bash
python3 classifier_trainer.py
\
python3 classifier_trainer.py
\
--mode
=
train_and_eval
\
--mode
=
train_and_eval
\
...
@@ -121,12 +125,15 @@ python3 classifier_trainer.py \
...
@@ -121,12 +125,15 @@ python3 classifier_trainer.py \
--config_file
=
configs/examples/resnet/imagenet/gpu.yaml
\
--config_file
=
configs/examples/resnet/imagenet/gpu.yaml
\
--params_override
=
'runtime.num_gpus=$NUM_GPUS'
--params_override
=
'runtime.num_gpus=$NUM_GPUS'
```
```
</details>
To train on multiple hosts, each with GPUs attached using
To train on multiple hosts, each with GPUs attached using
[
MultiWorkerMirroredStrategy
](
https://www.tensorflow.org/api_docs/python/tf/distribute/experimental/MultiWorkerMirroredStrategy
)
[
MultiWorkerMirroredStrategy
](
https://www.tensorflow.org/api_docs/python/tf/distribute/experimental/MultiWorkerMirroredStrategy
)
please update
`runtime`
section in gpu.yaml
please update
`runtime`
section in gpu.yaml
(or override using
`--params_override`
) with:
(or override using
`--params_override`
) with:
<details>
```
YAML
```
YAML
# gpu.yaml
# gpu.yaml
runtime:
runtime:
...
@@ -135,12 +142,16 @@ runtime:
...
@@ -135,12 +142,16 @@ runtime:
num_gpus: $NUM_GPUS
num_gpus: $NUM_GPUS
task_index: 0
task_index: 0
```
```
</details>
By having
`task_index: 0`
on the first host and
`task_index: 1`
on the second
By having
`task_index: 0`
on the first host and
`task_index: 1`
on the second
and so on.
`$HOST1`
and
`$HOST2`
are the IP addresses of the hosts, and
`port`
and so on.
`$HOST1`
and
`$HOST2`
are the IP addresses of the hosts, and
`port`
can be chosen any free port on the hosts. Only the first host will write
can be chosen any free port on the hosts. Only the first host will write
TensorBoard Summaries and save checkpoints.
TensorBoard Summaries and save checkpoints.
#### On TPU:
#### On TPU:
<details>
```
bash
```
bash
python3 classifier_trainer.py
\
python3 classifier_trainer.py
\
--mode
=
train_and_eval
\
--mode
=
train_and_eval
\
...
@@ -152,9 +163,13 @@ python3 classifier_trainer.py \
...
@@ -152,9 +163,13 @@ python3 classifier_trainer.py \
--config_file
=
configs/examples/resnet/imagenet/tpu.yaml
--config_file
=
configs/examples/resnet/imagenet/tpu.yaml
```
```
</details>
### VGG-16
### VGG-16
#### On GPU:
#### On GPU:
<details>
```
bash
```
bash
python3 classifier_trainer.py
\
python3 classifier_trainer.py
\
--mode
=
train_and_eval
\
--mode
=
train_and_eval
\
...
@@ -166,9 +181,13 @@ python3 classifier_trainer.py \
...
@@ -166,9 +181,13 @@ python3 classifier_trainer.py \
--params_override
=
'runtime.num_gpus=$NUM_GPUS'
--params_override
=
'runtime.num_gpus=$NUM_GPUS'
```
```
</details>
### EfficientNet
### EfficientNet
**Note: EfficientNet development is a work in progress.**
**Note: EfficientNet development is a work in progress.**
#### On GPU:
#### On GPU:
<details>
```
bash
```
bash
python3 classifier_trainer.py
\
python3 classifier_trainer.py
\
--mode
=
train_and_eval
\
--mode
=
train_and_eval
\
...
@@ -180,8 +199,11 @@ python3 classifier_trainer.py \
...
@@ -180,8 +199,11 @@ python3 classifier_trainer.py \
--params_override
=
'runtime.num_gpus=$NUM_GPUS'
--params_override
=
'runtime.num_gpus=$NUM_GPUS'
```
```
</details>
#### On TPU:
#### On TPU:
<details>
```
bash
```
bash
python3 classifier_trainer.py
\
python3 classifier_trainer.py
\
--mode
=
train_and_eval
\
--mode
=
train_and_eval
\
...
@@ -192,6 +214,7 @@ python3 classifier_trainer.py \
...
@@ -192,6 +214,7 @@ python3 classifier_trainer.py \
--data_dir
=
$DATA_DIR
\
--data_dir
=
$DATA_DIR
\
--config_file
=
configs/examples/efficientnet/imagenet/efficientnet-b0-tpu.yaml
--config_file
=
configs/examples/efficientnet/imagenet/efficientnet-b0-tpu.yaml
```
```
</details>
Note that the number of GPU devices can be overridden in the command line using
Note that the number of GPU devices can be overridden in the command line using
`--params_overrides`
. The TPU does not need this override as the device is fixed
`--params_overrides`
. The TPU does not need this override as the device is fixed
...
...
official/vision/beta/README.md
View file @
aa596195
This contents of
`beta`
folder is going to be deprecated soon and most of the
Contents of this
`beta`
folder is going to be deprecated soon and most of the
content has been moved to[official/vision]
content has been moved to
[
official/vision
](
https://github.com/tensorflow/models/tree/master/official/vision
)
.
(https://github.com/tensorflow/models/tree/master/official/vision).
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