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)