Fix the batch_size of the ncf benchmark (tensorflow#6597)

EdisonAltamirano · Apr 18, 2019 · f519c01 · f519c01
1 parent c33d3ef
commit f519c01
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/official/recommendation/ncf_keras_benchmark.py b/official/recommendation/ncf_keras_benchmark.py
@@ -86,7 +86,7 @@ def __init__(self,
     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 @@ def __init__(self,
     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