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)