Skip to content

Commit fb23bcc

Browse files
committed
adjust log format and tf verbosity
1 parent c30c692 commit fb23bcc

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

tf2onnx/verbose_logging.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ def getLogger(name=None): # pylint: disable=invalid-name, function-redefined
3232
return logger
3333

3434

35-
_SIMPLE_LOG_FORMAT = "%(message)s"
35+
_BASIC_LOG_FORMAT = "%(asctime)s - %(levelname)s - %(message)s"
3636
_VERBOSE_LOG_FORMAT = "%(asctime)s - %(levelname)s - %(name)s: %(message)s"
3737

3838

@@ -41,7 +41,7 @@ def basicConfig(**kwargs): # pylint: disable=invalid-name, function-redefined
4141
# Choose pre-defined format if format argument is not specified
4242
if "format" not in kwargs:
4343
level = kwargs.get("level", _logging.root.level)
44-
kwargs["format"] = _SIMPLE_LOG_FORMAT if level >= INFO else _VERBOSE_LOG_FORMAT
44+
kwargs["format"] = _BASIC_LOG_FORMAT if level >= INFO else _VERBOSE_LOG_FORMAT
4545

4646
_logging.basicConfig(**kwargs)
4747
set_tf_verbosity(_logging.getLogger().getEffectiveLevel())
@@ -67,8 +67,7 @@ def set_level(level):
6767
def set_tf_verbosity(level):
6868
""" Set TF logging verbosity."""
6969
# TF log is too verbose, adjust it
70-
level = WARNING if level >= VERBOSE else level
71-
70+
level = ERROR if level >= INFO else level
7271
tf.logging.set_verbosity(level)
7372

7473
# TF_CPP_MIN_LOG_LEVEL:

0 commit comments

Comments
 (0)