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
2662da2c
"cpp_onnx/src/FeatureQueue.cpp" did not exist on "06924f5d520d11f755f7fd401b2b04d4a1db003b"
Commit
2662da2c
authored
Feb 26, 2018
by
Suharsh Sivakumar
Browse files
improve docs and make train flag match eval flag
parent
1abcb9c3
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
9 additions
and
7 deletions
+9
-7
research/slim/nets/mobilenet_v1.md
research/slim/nets/mobilenet_v1.md
+6
-5
research/slim/nets/mobilenet_v1_train.py
research/slim/nets/mobilenet_v1_train.py
+3
-2
No files found.
research/slim/nets/mobilenet_v1.md
View file @
2662da2c
...
@@ -96,13 +96,13 @@ $ bazel build -c opt --config=cuda mobilenet_v1_{eval,train}
...
@@ -96,13 +96,13 @@ $ bazel build -c opt --config=cuda mobilenet_v1_{eval,train}
Train:
Train:
```
```
$ ./bazel-bin/mobilenet_v1_train
$ ./bazel-bin/mobilenet_v1_train
--dataset_dir "path/to/dataset" --checkpoint_dir "path/to/checkpoints"
```
```
Eval:
Eval:
```
```
$ ./bazel-bin/mobilenet_v1_eval
$ ./bazel-bin/mobilenet_v1_eval
--dataset_dir "path/to/dataset" --checkpoint_dir "path/to/checkpoints"
```
```
#### Quantized Training and Eval
#### Quantized Training and Eval
...
@@ -110,19 +110,20 @@ $ ./bazel-bin/mobilenet_v1_eval
...
@@ -110,19 +110,20 @@ $ ./bazel-bin/mobilenet_v1_eval
Train from preexisting float checkpoint:
Train from preexisting float checkpoint:
```
```
$ ./bazel-bin/mobilenet_v1_train --quantize=True --fine_tune_checkpoint=checkpoint-name
$ ./bazel-bin/mobilenet_v1_train --dataset_dir "path/to/dataset" --checkpoint_dir "path/to/checkpoints" \
--quantize=True --fine_tune_checkpoint=float/checkpoint/path
```
```
Train from scratch:
Train from scratch:
```
```
$ ./bazel-bin/mobilenet_v1_train --quantize=True
$ ./bazel-bin/mobilenet_v1_train
--dataset_dir "path/to/dataset" --checkpoint_dir "path/to/checkpoints"
--quantize=True
```
```
Eval:
Eval:
```
```
$ ./bazel-bin/mobilenet_v1_eval --quantize=True
$ ./bazel-bin/mobilenet_v1_eval
--dataset_dir "path/to/dataset" --checkpoint_dir "path/to/checkpoints"
--quantize=True
```
```
The resulting float and quantized models can be run on-device via
[
TensorFlow Lite
](
https://www.tensorflow.org/mobile/tflite/
)
.
The resulting float and quantized models can be run on-device via
[
TensorFlow Lite
](
https://www.tensorflow.org/mobile/tflite/
)
.
...
...
research/slim/nets/mobilenet_v1_train.py
View file @
2662da2c
...
@@ -40,7 +40,8 @@ flags.DEFINE_float('depth_multiplier', 1.0, 'Depth multiplier for mobilenet')
...
@@ -40,7 +40,8 @@ flags.DEFINE_float('depth_multiplier', 1.0, 'Depth multiplier for mobilenet')
flags
.
DEFINE_bool
(
'quantize'
,
False
,
'Quantize training'
)
flags
.
DEFINE_bool
(
'quantize'
,
False
,
'Quantize training'
)
flags
.
DEFINE_string
(
'fine_tune_checkpoint'
,
''
,
flags
.
DEFINE_string
(
'fine_tune_checkpoint'
,
''
,
'Checkpoint from which to start finetuning.'
)
'Checkpoint from which to start finetuning.'
)
flags
.
DEFINE_string
(
'logdir'
,
''
,
'Directory for writing training event logs'
)
flags
.
DEFINE_string
(
'checkpoint_dir'
,
''
,
'Directory for writing training checkpoints and logs'
)
flags
.
DEFINE_string
(
'dataset_dir'
,
''
,
'Location of dataset'
)
flags
.
DEFINE_string
(
'dataset_dir'
,
''
,
'Location of dataset'
)
flags
.
DEFINE_integer
(
'log_every_n_steps'
,
100
,
'Number of steps per log'
)
flags
.
DEFINE_integer
(
'log_every_n_steps'
,
100
,
'Number of steps per log'
)
flags
.
DEFINE_integer
(
'save_summaries_secs'
,
100
,
flags
.
DEFINE_integer
(
'save_summaries_secs'
,
100
,
...
@@ -191,7 +192,7 @@ def train_model():
...
@@ -191,7 +192,7 @@ def train_model():
with
g
.
as_default
():
with
g
.
as_default
():
slim
.
learning
.
train
(
slim
.
learning
.
train
(
train_tensor
,
train_tensor
,
FLAGS
.
log
dir
,
FLAGS
.
checkpoint_
dir
,
is_chief
=
(
FLAGS
.
task
==
0
),
is_chief
=
(
FLAGS
.
task
==
0
),
master
=
FLAGS
.
master
,
master
=
FLAGS
.
master
,
log_every_n_steps
=
FLAGS
.
log_every_n_steps
,
log_every_n_steps
=
FLAGS
.
log_every_n_steps
,
...
...
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