def register_callback():
from polygraphy.logger.logger import G_LOGGER
def set_tf_logging_level(sev):
import os
import tensorflow as tf
if sev > G_LOGGER.WARNING:
tf_sev = tf.compat.v1.logging.ERROR
tf_logging_level = "3"
elif sev > G_LOGGER.INFO:
tf_sev = tf.compat.v1.logging.WARN
tf_logging_level = "2"
elif sev > G_LOGGER.VERBOSE:
tf_sev = tf.compat.v1.logging.INFO
tf_logging_level = "1"
else:
tf_sev = tf.compat.v1.logging.DEBUG
tf_logging_level = "0"
tf.compat.v1.logging.set_verbosity(tf_sev)
os.environ['TF_CPP_MIN_LOG_LEVEL'] = tf_logging_level
G_LOGGER.register_callback(
set_tf_logging_level
) # Will be registered when this runner is imported.
def register_callback():
from polygraphy.logger.logger import G_LOGGER
def set_onnx_logging_level(sev):
import warnings
if sev >= G_LOGGER.INFO:
warnings.filterwarnings("ignore")
else:
warnings.filterwarnings("default")
G_LOGGER.register_callback(set_onnx_logging_level)
def register_callback():
from polygraphy.logger.logger import G_LOGGER
def set_trt_logging_level(sev):
import tensorrt as trt
if sev >= G_LOGGER.CRITICAL:
TRT_LOGGER.min_severity = trt.Logger.INTERNAL_ERROR
elif sev >= G_LOGGER.ERROR:
TRT_LOGGER.min_severity = trt.Logger.ERROR
elif sev >= G_LOGGER.INFO:
TRT_LOGGER.min_severity = trt.Logger.WARNING
elif sev >= G_LOGGER.VERBOSE:
TRT_LOGGER.min_severity = trt.Logger.INFO
else:
TRT_LOGGER.min_severity = trt.Logger.VERBOSE
G_LOGGER.register_callback(set_trt_logging_level) # Will be registered when this runner is imported.
