torch_tensorrt.logging¶
- class torch_tensorrt.logging.debug[source]¶
上下文管理器,用于通过日志记录器显示完整的调试信息
示例
with torch_tensorrt.logging.debug(): model_trt = torch_tensorrt.compile(model, **spec)
- class torch_tensorrt.logging.errors[source]¶
上下文管理器,用于将显示的日志消息限制为仅错误及以上级别
示例
with torch_tensorrt.logging.errors(): outputs = model_torchtrt(inputs)
- class torch_tensorrt.logging.graphs[source]¶
上下文管理器,用于显示中间降低过程的结果以及通过日志记录器的完整调试信息
示例
with torch_tensorrt.logging.graphs(): model_trt = torch_tensorrt.compile(model, **spec)
- class torch_tensorrt.logging.info[source]¶
上下文管理器,用于显示所有信息及更高严重性级别消息
示例
with torch_tensorrt.logging.info(): model_trt = torch_tensorrt.compile(model, **spec)