快捷方式

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)
class torch_tensorrt.logging.internal_errors[source]

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

示例

with torch_tensorrt.logging.internal_errors():
    outputs = model_torchtrt(inputs)
class torch_tensorrt.logging.warnings[source]

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

示例

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

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并解答您的问题

查看资源