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
1157c738
"...git@developer.sourcefind.cn:OpenDAS/torch-harmonics.git" did not exist on "24fcb06e690f9ab7d036851b94a4dde9bf0e8b54"
Commit
1157c738
authored
Jun 24, 2019
by
nnigania
Committed by
Toby Boyd
Jun 24, 2019
Browse files
adding 8 gpu test for ncf (#7092)
parent
adc27172
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
17 additions
and
0 deletions
+17
-0
official/recommendation/ncf_keras_benchmark.py
official/recommendation/ncf_keras_benchmark.py
+17
-0
No files found.
official/recommendation/ncf_keras_benchmark.py
View file @
1157c738
...
@@ -250,6 +250,23 @@ class NCFKerasAccuracy(NCFKerasBenchmarkBase):
...
@@ -250,6 +250,23 @@ class NCFKerasAccuracy(NCFKerasBenchmarkBase):
FLAGS
.
epsilon
=
1e-8
FLAGS
.
epsilon
=
1e-8
self
.
_run_and_report_benchmark
()
self
.
_run_and_report_benchmark
()
def
benchmark_8_gpu_mlperf_like
(
self
):
"""8 GPU test meant to compare Google implementation
with MLperf top line submission using the
hyper-parameters from the winning MLPerf0.5 submission.
Using similar rules as MLPerf0.5
Fixed epochs to MLPerf sumbmission's convergnce on 17 epochs
"""
self
.
_setup
()
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
()
class
NCFKerasSynth
(
NCFKerasBenchmarkBase
):
class
NCFKerasSynth
(
NCFKerasBenchmarkBase
):
"""Benchmark NCF model using synthetic data."""
"""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