Commit 1f9efe10 authored by A. Unique TensorFlower's avatar A. Unique TensorFlower
Browse files

change benchmark's log verbosity to logging.INFO. it seems to me that DEBUG...

change benchmark's log verbosity to logging.INFO. it seems to me that DEBUG map to ---v=1 internally, which is way to verbose for the purpose of benchmarking.

PiperOrigin-RevId: 274040907
parent b63030a3
...@@ -56,7 +56,7 @@ class EstimatorBenchmark(tf.test.Benchmark): ...@@ -56,7 +56,7 @@ class EstimatorBenchmark(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."""
tf.compat.v1.logging.set_verbosity(tf.compat.v1.logging.DEBUG) tf.compat.v1.logging.set_verbosity(tf.compat.v1.logging.INFO)
if EstimatorBenchmark.local_flags is None: if EstimatorBenchmark.local_flags is None:
for flag_method in self.flag_methods: for flag_method in self.flag_methods:
flag_method() flag_method()
......
...@@ -48,7 +48,7 @@ class NCFKerasBenchmarkBase(tf.test.Benchmark): ...@@ -48,7 +48,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.') 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.INFO)
if NCFKerasBenchmarkBase.local_flags is None: if NCFKerasBenchmarkBase.local_flags is None:
ncf_common.define_ncf_flags() ncf_common.define_ncf_flags()
# Loads flags to get defaults to then override. List cannot be empty. # Loads flags to get defaults to then override. List cannot be empty.
......
...@@ -54,7 +54,7 @@ class EstimatorBenchmark(tf.test.Benchmark): ...@@ -54,7 +54,7 @@ class EstimatorBenchmark(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."""
tf.compat.v1.logging.set_verbosity(tf.compat.v1.logging.DEBUG) tf.compat.v1.logging.set_verbosity(tf.compat.v1.logging.INFO)
if EstimatorBenchmark.local_flags is None: if EstimatorBenchmark.local_flags is None:
for flag_method in self.flag_methods: for flag_method in self.flag_methods:
flag_method() flag_method()
......
...@@ -55,7 +55,7 @@ class PerfZeroBenchmark(tf.test.Benchmark): ...@@ -55,7 +55,7 @@ class PerfZeroBenchmark(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."""
tf.compat.v1.logging.set_verbosity(tf.compat.v1.logging.DEBUG) tf.compat.v1.logging.set_verbosity(tf.compat.v1.logging.INFO)
if PerfZeroBenchmark.local_flags is None: if PerfZeroBenchmark.local_flags is None:
for flag_method in self.flag_methods: for flag_method in self.flag_methods:
flag_method() flag_method()
......
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