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
3c30aa56
Commit
3c30aa56
authored
Aug 13, 2019
by
Hongkun Yu
Committed by
A. Unique TensorFlower
Aug 13, 2019
Browse files
Internal change
PiperOrigin-RevId: 263217684
parent
cce82cd0
Changes
1
Show 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 @
3c30aa56
...
...
@@ -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