Internal change

PiperOrigin-RevId: 266002056

Internal change
PiperOrigin-RevId: 266002056
503b1e61 · David Chen · A. Unique TensorFlower · 6def0d03 · 503b1e61
Commit 503b1e61 authored Aug 28, 2019 by David Chen Committed by A. Unique TensorFlower Aug 28, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

official/recommendation/ncf_keras_benchmark.py official/recommendation/ncf_keras_benchmark.py +4 -0

No files found.
--- a/official/recommendation/ncf_keras_benchmark.py
+++ b/official/recommendation/ncf_keras_benchmark.py
@@ -376,3 +376,7 @@ class NCFKerasSynth(NCFKerasBenchmarkBase):
    self._setup()
    FLAGS.num_gpus = 2
    self._run_and_report_benchmark()
+
+
+if __name__ == '__main__':
+  tf.test.main()