Skip to content

GitLab

Commit 74d924e9 authored by Shining Sun's avatar Shining Sun Committed by Toby Boyd
Browse files

added missing flags to the real data benchmark (#6483)

parent 371645fc
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
official/recommendation/ncf_keras_benchmark.py
View file @ 74d924e9
......@@ -85,6 +85,7 @@ class KerasNCFRealData(KerasNCFBenchmarkBase):
default_flags['dataset'] = 'ml-20m'
default_flags['num_gpus'] = 1
default_flags['train_epochs'] = 14
default_flags['clean'] = True
default_flags['batch_size'] = 16000
default_flags['learning_rate'] = 0.00382059
default_flags['beta1'] = 0.783529
......@@ -93,6 +94,7 @@ class KerasNCFRealData(KerasNCFBenchmarkBase):
default_flags['layers'] = [256, 256, 128, 64]
default_flags['num_factors'] = 64
default_flags['hr_threshold'] = 0.635
default_flags['ml_perf'] = True
default_flags['use_synthetic_data'] = False
super(KerasNCFRealData, self).__init__(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment