Skip to content

GitLab

"git@developer.sourcefind.cn:modelzoo/resnet50_tensorflow.git" did not exist on "c115444fa5966d95c64ec1ecab89e422a9477a1c"
Commit 503b1e61 authored by David Chen's avatar David Chen Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 266002056
parent 6def0d03
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
official/recommendation/ncf_keras_benchmark.py
View file @ 503b1e61
...@@ -376,3 +376,7 @@ class NCFKerasSynth(NCFKerasBenchmarkBase): ...@@ -376,3 +376,7 @@ class NCFKerasSynth(NCFKerasBenchmarkBase):
self._setup() self._setup()
FLAGS.num_gpus = 2 FLAGS.num_gpus = 2
self._run_and_report_benchmark() self._run_and_report_benchmark()
if __name__ == '__main__':
tf.test.main()
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