Skip to content

GitLab

Commit bb328876 authored by Adrian Kuegel's avatar Adrian Kuegel Committed by A. Unique TensorFlower
Browse files

Adjust ncf_keras_benchmark to force TF 2.0 when running internally. 

PiperOrigin-RevId: 267946336
parent 64f87cd2
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
official/recommendation/ncf_keras_benchmark.py
View file @ bb328876
...@@ -47,6 +47,7 @@ class NCFKerasBenchmarkBase(tf.test.Benchmark): ...@@ -47,6 +47,7 @@ class NCFKerasBenchmarkBase(tf.test.Benchmark):
def _setup(self): def _setup(self):
"""Sets up and resets flags before each test.""" """Sets up and resets flags before each test."""
assert tf.version.VERSION.startswith('2.')
tf.compat.v1.logging.set_verbosity(tf.compat.v1.logging.DEBUG) tf.compat.v1.logging.set_verbosity(tf.compat.v1.logging.DEBUG)
if NCFKerasBenchmarkBase.local_flags is None: if NCFKerasBenchmarkBase.local_flags is None:
ncf_common.define_ncf_flags() ncf_common.define_ncf_flags()
......
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