torch_tensorrt.logging

class torch_tensorrt.logging.debug

用于通过日志记录器显示完整调试信息的上下文管理器

示例

with torch_tensorrt.logging.debug():
    model_trt = torch_tensorrt.compile(model, **spec)

class torch_tensorrt.logging.errors

用于将显示的日志消息限制为仅错误及以上级别的上下文管理器

示例

with torch_tensorrt.logging.errors():
    outputs = model_torchtrt(inputs)

class torch_tensorrt.logging.graphs

用于通过日志记录器显示中间降级过程的结果以及完整调试信息的上下文管理器

示例

with torch_tensorrt.logging.graphs():
    model_trt = torch_tensorrt.compile(model, **spec)

class torch_tensorrt.logging.info

用于显示所有信息及更高严重性级别消息的上下文管理器

示例

with torch_tensorrt.logging.info():
    model_trt = torch_tensorrt.compile(model, **spec)

class torch_tensorrt.logging.internal_errors

用于将显示的日志消息限制为仅内部错误的上下文管理器

示例

with torch_tensorrt.logging.internal_errors():
    outputs = model_torchtrt(inputs)

class torch_tensorrt.logging.warnings

用于将显示的日志消息限制为仅警告及以上级别的上下文管理器

示例

with torch_tensorrt.logging.warnings():
    model_trt = torch_tensorrt.compile(model, **spec)