Fix the batch_size of the ncf benchmark (#6597)

f519c015 · Shining Sun · GitHub · c33d3ef4 · f519c015
Unverified Commit f519c015 authored Apr 18, 2019 by Shining Sun Committed by GitHub Apr 18, 2019
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

official/recommendation/ncf_keras_benchmark.py official/recommendation/ncf_keras_benchmark.py +2 -2

No files found.
--- a/official/recommendation/ncf_keras_benchmark.py
+++ b/official/recommendation/ncf_keras_benchmark.py
@@ -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