🐛 fix tensorflow logger handler assignment

This commit is contained in:
Faylixe
2020-12-08 15:16:34 +01:00
parent 914f4e01ef
commit f5deb8cb85

View File

@@ -42,8 +42,8 @@ def configure_logger(verbose: bool) -> None:
verbose (bool):
`True` to use verbose logger, `False` otherwise.
"""
# tf_logger = tf_logging._get_logger()
# tf_logger.handlers = [handler]
tf_logger = tf_logging.get_logger()
tf_logger.handlers = [handler]
if verbose:
environ['TF_CPP_MIN_LOG_LEVEL'] = '1'
tf_logging.set_verbosity(tf_logging.INFO)