加载器

模块: polygraphy.tools.args

class OnnxrtSessionArgs[source]

基类: BaseArgs

ONNX-Runtime 会话创建: 创建 ONNX-Runtime 推理会话

依赖于

  • OnnxLoadArgs

  • ModelArgs

parse_impl(args)[source]

解析命令行参数并填充以下属性

providers

执行提供程序的列表。

类型:

List[str]

load_onnxrt_session(model=None)[source]

根据命令行上提供的参数加载 ONNX-Runtime 推理会话。

参数:

model (Union[bytes, str]) – 模型字节或模型路径。 默认为 None，在这种情况下，将使用命令行上指定的模型。

返回:

onnxruntime.InferenceSession