def register_logger_callback(): from polygraphy.logger import G_LOGGER def set_tf_logging_level(sev): import os from polygraphy import mod tf = mod.lazy_import("tensorflow", version="<2.0") if not mod.has_mod(tf, with_attr="__version__"): return 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 run(self, args): def set_onnx_gs_logging_level(sev): ONNX_GS_LOGGER = gs.logger.G_LOGGER if sev >= G_LOGGER.CRITICAL: ONNX_GS_LOGGER.severity = ONNX_GS_LOGGER.CRITICAL elif sev >= G_LOGGER.ERROR: ONNX_GS_LOGGER.severity = ONNX_GS_LOGGER.ERROR elif sev >= G_LOGGER.WARNING: ONNX_GS_LOGGER.severity = ONNX_GS_LOGGER.WARNING elif sev >= G_LOGGER.INFO: ONNX_GS_LOGGER.severity = ONNX_GS_LOGGER.INFO elif sev >= G_LOGGER.EXTRA_VERBOSE: ONNX_GS_LOGGER.severity = ONNX_GS_LOGGER.DEBUG elif sev >= G_LOGGER.SUPER_VERBOSE: ONNX_GS_LOGGER.severity = ONNX_GS_LOGGER.VERBOSE else: ONNX_GS_LOGGER.severity = ONNX_GS_LOGGER.ULTRA_VERBOSE fmts = self.arg_groups[LoggerArgs].log_format for fmt in fmts: if fmt == "no-colors": ONNX_GS_LOGGER.colors = False elif fmt == "timestamp": ONNX_GS_LOGGER.timestamp = True elif fmt == "line-info": ONNX_GS_LOGGER.line_info = True G_LOGGER.register_callback(set_onnx_gs_logging_level) return self.run_impl(args)
def register_logger_callback(): from polygraphy.logger import G_LOGGER def set_trt_logging_level(sev): from polygraphy import mod trt = mod.lazy_import("tensorrt") if not mod.has_mod(trt, with_attr="__version__"): return if sev >= G_LOGGER.CRITICAL: get_trt_logger().min_severity = trt.Logger.INTERNAL_ERROR elif sev >= G_LOGGER.ERROR: get_trt_logger().min_severity = trt.Logger.ERROR elif sev >= G_LOGGER.INFO: get_trt_logger().min_severity = trt.Logger.WARNING elif sev >= G_LOGGER.VERBOSE: get_trt_logger().min_severity = trt.Logger.INFO else: get_trt_logger().min_severity = trt.Logger.VERBOSE G_LOGGER.register_callback( set_trt_logging_level ) # Will be registered when this runner is imported.