From 1f9efe10556ae20f09eddb54b9d035fc3aceed71 Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Thu, 10 Oct 2019 14:34:47 -0700 Subject: [PATCH] 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 --- official/r1/resnet/estimator_benchmark.py | 2 +- official/recommendation/ncf_keras_benchmark.py | 2 +- official/transformer/transformer_estimator_benchmark.py | 2 +- official/utils/testing/perfzero_benchmark.py | 2 +- 4 files changed, 4 insertions(+), 4 deletions(-) diff --git a/official/r1/resnet/estimator_benchmark.py b/official/r1/resnet/estimator_benchmark.py index c0e4cdb1b..ce28b3c29 100644 --- a/official/r1/resnet/estimator_benchmark.py +++ b/official/r1/resnet/estimator_benchmark.py @@ -56,7 +56,7 @@ class EstimatorBenchmark(tf.test.Benchmark): def _setup(self): """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: for flag_method in self.flag_methods: flag_method() diff --git a/official/recommendation/ncf_keras_benchmark.py b/official/recommendation/ncf_keras_benchmark.py index e1a3d4912..718b49ea4 100644 --- a/official/recommendation/ncf_keras_benchmark.py +++ b/official/recommendation/ncf_keras_benchmark.py @@ -48,7 +48,7 @@ class NCFKerasBenchmarkBase(tf.test.Benchmark): def _setup(self): """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.INFO) if NCFKerasBenchmarkBase.local_flags is None: ncf_common.define_ncf_flags() # Loads flags to get defaults to then override. List cannot be empty. diff --git a/official/transformer/transformer_estimator_benchmark.py b/official/transformer/transformer_estimator_benchmark.py index 618f27d5a..f1b757b04 100644 --- a/official/transformer/transformer_estimator_benchmark.py +++ b/official/transformer/transformer_estimator_benchmark.py @@ -54,7 +54,7 @@ class EstimatorBenchmark(tf.test.Benchmark): def _setup(self): """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: for flag_method in self.flag_methods: flag_method() diff --git a/official/utils/testing/perfzero_benchmark.py b/official/utils/testing/perfzero_benchmark.py index 7ec14eeaf..445c04597 100644 --- a/official/utils/testing/perfzero_benchmark.py +++ b/official/utils/testing/perfzero_benchmark.py @@ -55,7 +55,7 @@ class PerfZeroBenchmark(tf.test.Benchmark): def _setup(self): """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: for flag_method in self.flag_methods: flag_method() -- GitLab