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
b48f4db6
Commit
b48f4db6
authored
Jan 29, 2020
by
Yanhui Liang
Committed by
A. Unique TensorFlower
Jan 29, 2020
Browse files
Internal change
PiperOrigin-RevId: 292159895
parent
ddc40272
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
8 additions
and
0 deletions
+8
-0
official/benchmark/bert_squad_benchmark.py
official/benchmark/bert_squad_benchmark.py
+8
-0
No files found.
official/benchmark/bert_squad_benchmark.py
View file @
b48f4db6
...
@@ -494,6 +494,14 @@ class BertSquadMultiWorkerAccuracy(BertSquadBenchmarkBase):
...
@@ -494,6 +494,14 @@ class BertSquadMultiWorkerAccuracy(BertSquadBenchmarkBase):
self
.
_run_and_report_benchmark
()
self
.
_run_and_report_benchmark
()
def
benchmark_eager_8_gpu_2_workers_fp16_ring_tweaked
(
self
):
"""8 GPUs per worker, 2 workers, fp16, ring all-reduce."""
self
.
_benchmark_common
(
num_workers
=
2
,
all_reduce_alg
=
'ring'
)
def
benchmark_eager_8_gpu_2_workers_fp16_nccl_tweaked
(
self
):
"""8 GPUs per worker, 2 workers, fp16, nccl all-reduce."""
self
.
_benchmark_common
(
num_workers
=
2
,
all_reduce_alg
=
'nccl'
)
def
benchmark_8_gpu_8_workers_fp16_ring_tweaked
(
self
):
def
benchmark_8_gpu_8_workers_fp16_ring_tweaked
(
self
):
"""8 GPUs per worker, 8 workers, fp16, ring all-reduce."""
"""8 GPUs per worker, 8 workers, fp16, ring all-reduce."""
self
.
_benchmark_common
(
num_workers
=
8
,
all_reduce_alg
=
'ring'
)
self
.
_benchmark_common
(
num_workers
=
8
,
all_reduce_alg
=
'ring'
)
...
...
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