Shortcuts

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)