Skip to content

GitLab

Unverified Commit f519c015 authored by Shining Sun's avatar Shining Sun Committed by GitHub
Browse files

Fix the batch_size of the ncf benchmark (#6597)

parent c33d3ef4
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
official/recommendation/ncf_keras_benchmark.py
View file @ f519c015
...@@ -86,7 +86,7 @@ class KerasNCFRealData(KerasNCFBenchmarkBase): ...@@ -86,7 +86,7 @@ class KerasNCFRealData(KerasNCFBenchmarkBase):
default_flags['num_gpus'] = 1 default_flags['num_gpus'] = 1
default_flags['train_epochs'] = 14 default_flags['train_epochs'] = 14
default_flags['clean'] = True default_flags['clean'] = True
default_flags['batch_size'] = 16000 default_flags['batch_size'] = 160000
default_flags['learning_rate'] = 0.00382059 default_flags['learning_rate'] = 0.00382059
default_flags['beta1'] = 0.783529 default_flags['beta1'] = 0.783529
default_flags['beta2'] = 0.909003 default_flags['beta2'] = 0.909003
...@@ -126,7 +126,7 @@ class KerasNCFSyntheticData(KerasNCFBenchmarkBase): ...@@ -126,7 +126,7 @@ class KerasNCFSyntheticData(KerasNCFBenchmarkBase):
default_flags['dataset'] = 'ml-20m' default_flags['dataset'] = 'ml-20m'
default_flags['num_gpus'] = 1 default_flags['num_gpus'] = 1
default_flags['train_epochs'] = 14 default_flags['train_epochs'] = 14
default_flags['batch_size'] = 16000 default_flags['batch_size'] = 160000
default_flags['learning_rate'] = 0.00382059 default_flags['learning_rate'] = 0.00382059
default_flags['beta1'] = 0.783529 default_flags['beta1'] = 0.783529
default_flags['beta2'] = 0.909003 default_flags['beta2'] = 0.909003
......
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