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
Hide 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):
default_flags['num_gpus'] = 1
default_flags['train_epochs'] = 14
default_flags['clean'] = True
default_flags['batch_size'] = 16000
default_flags['batch_size'] = 160000
default_flags['learning_rate'] = 0.00382059
default_flags['beta1'] = 0.783529
default_flags['beta2'] = 0.909003
......@@ -126,7 +126,7 @@ class KerasNCFSyntheticData(KerasNCFBenchmarkBase):
default_flags['dataset'] = 'ml-20m'
default_flags['num_gpus'] = 1
default_flags['train_epochs'] = 14
default_flags['batch_size'] = 16000
default_flags['batch_size'] = 160000
default_flags['learning_rate'] = 0.00382059
default_flags['beta1'] = 0.783529
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