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
01d28a37
"vscode:/vscode.git/clone" did not exist on "1b2e64cd75c8d1c238c61a3bc1e3c62a7d403b53"
Commit
01d28a37
authored
Aug 13, 2019
by
Toby Boyd
Committed by
Hongkun Yu
Aug 13, 2019
Browse files
Remove 8xGPU XLA tests. They are not supported yet. (#7435)
parent
e36934b3
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
0 additions
and
27 deletions
+0
-27
official/recommendation/ncf_keras_benchmark.py
official/recommendation/ncf_keras_benchmark.py
+0
-27
No files found.
official/recommendation/ncf_keras_benchmark.py
View file @
01d28a37
...
...
@@ -302,19 +302,6 @@ class NCFKerasAccuracy(NCFKerasBenchmarkBase):
FLAGS
.
force_v2_in_keras_compile
=
False
self
.
_run_and_report_benchmark_mlperf_like
()
def
benchmark_xla_8_gpu_mlperf_like
(
self
):
"""8 GPU using keras fit/compile with XLA."""
self
.
_setup
()
FLAGS
.
num_gpus
=
8
FLAGS
.
enable_xla
=
True
FLAGS
.
train_epochs
=
17
FLAGS
.
batch_size
=
1048576
FLAGS
.
learning_rate
=
0.0045
FLAGS
.
beta1
=
0.25
FLAGS
.
beta2
=
0.5
FLAGS
.
epsilon
=
1e-8
self
.
_run_and_report_benchmark_mlperf_like
()
def
benchmark_8_gpu_ctl_mlperf_like
(
self
):
"""8 GPU using CTL."""
self
.
_setup
()
...
...
@@ -328,20 +315,6 @@ class NCFKerasAccuracy(NCFKerasBenchmarkBase):
FLAGS
.
epsilon
=
1e-8
self
.
_run_and_report_benchmark_mlperf_like
()
def
benchmark_xla_8_gpu_ctl_mlperf_like
(
self
):
"""8 GPU using CTL with XLA."""
self
.
_setup
()
FLAGS
.
keras_use_ctl
=
True
FLAGS
.
enable_xla
=
True
FLAGS
.
num_gpus
=
8
FLAGS
.
train_epochs
=
17
FLAGS
.
batch_size
=
1048576
FLAGS
.
learning_rate
=
0.0045
FLAGS
.
beta1
=
0.25
FLAGS
.
beta2
=
0.5
FLAGS
.
epsilon
=
1e-8
self
.
_run_and_report_benchmark_mlperf_like
()
class
NCFKerasSynth
(
NCFKerasBenchmarkBase
):
"""Benchmark NCF model using synthetic data."""
...
...
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