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
ddc40272
Commit
ddc40272
authored
Jan 28, 2020
by
Sai Ganesh Bandiatmakuri
Committed by
A. Unique TensorFlower
Jan 28, 2020
Browse files
Internal change
PiperOrigin-RevId: 292030392
parent
4f3acbda
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
23 additions
and
0 deletions
+23
-0
official/benchmark/resnet_ctl_imagenet_benchmark.py
official/benchmark/resnet_ctl_imagenet_benchmark.py
+23
-0
No files found.
official/benchmark/resnet_ctl_imagenet_benchmark.py
View file @
ddc40272
...
...
@@ -318,6 +318,29 @@ class Resnet50CtlBenchmarkBase(CtlBenchmark):
FLAGS
.
dtype
=
'fp16'
self
.
_run_and_report_benchmark
()
def
benchmark_8_gpu_eager
(
self
):
"""Test Keras model with 8 GPUs, eager, fp32."""
self
.
_setup
()
FLAGS
.
num_gpus
=
8
FLAGS
.
use_tf_function
=
False
FLAGS
.
distribution_strategy
=
'mirrored'
FLAGS
.
model_dir
=
self
.
_get_model_dir
(
'benchmark_8_gpu_eager'
)
FLAGS
.
batch_size
=
128
self
.
_run_and_report_benchmark
()
def
benchmark_8_gpu_eager_fp16
(
self
):
"""Test Keras model with 8 GPUs, eager, fp16."""
self
.
_setup
()
FLAGS
.
num_gpus
=
8
FLAGS
.
dtype
=
'fp16'
FLAGS
.
use_tf_function
=
False
FLAGS
.
distribution_strategy
=
'mirrored'
FLAGS
.
model_dir
=
self
.
_get_model_dir
(
'benchmark_8_gpu_eager_fp16'
)
FLAGS
.
batch_size
=
128
self
.
_run_and_report_benchmark
()
def
benchmark_8_gpu_amp
(
self
):
"""Test Keras model with 8 GPUs with automatic mixed precision."""
self
.
_setup
()
...
...
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