OnnxOptions 类

定义

命名空间:

Microsoft.ML.Transforms.Onnx

程序集:

Microsoft.ML.OnnxTransformer.dll

包:

Microsoft.ML.OnnxTransformer v3.0.1

的选项 OnnxScoringEstimator。

public sealed class OnnxOptions

type OnnxOptions = class

Public NotInheritable Class OnnxOptions

继承

Object

OnnxOptions

构造函数

OnnxOptions()

的选项 OnnxScoringEstimator。

字段

FallbackToCpu	如果为 true，则 GPU 出错时继续在 CPU 上执行。 如果为 false， 将引发 GPU 异常。
GpuDeviceId	在 (上运行的 GPU 设备 ID，例如 0,1,..) 。 CPU 为 Null。 需要 CUDA 10.1。
InputColumns	输入列的名称。
InterOpNumThreads	控制用于跨节点) 并行执行图形 (的线程数。
IntraOpNumThreads	控制用于运行模型的线程数。
ModelFile	onnx 模型文件的路径。
OutputColumns	输出列的名称。
RecursionLimit	Protobuf CodedInputStream 递归限制。
ShapeDictionary	要用于从 ModelFile加载的 ONNX 形状。