OnnxOptions Class

Definition

The options for an OnnxScoringEstimator.

public sealed class OnnxOptions
type OnnxOptions = class
Public NotInheritable Class OnnxOptions
Inheritance
OnnxOptions

Constructors

OnnxOptions()

Fields

FallbackToCpu

If true, resumes execution on CPU upon GPU error. If false, will raise the GPU exception.

GpuDeviceId

GPU device id to run on (e.g. 0,1,..). Null for CPU. Requires CUDA 10.1.

InputColumns

Name of the input column.

InterOpNumThreads

Controls the number of threads used to parallelize the execution of the graph (across nodes).

IntraOpNumThreads

Controls the number of threads to use to run the model.

ModelFile

Path to the onnx model file.

OutputColumns

Name of the output column.

RecursionLimit

Protobuf CodedInputStream recursion limit.

ShapeDictionary

ONNX shapes to be used over those loaded from ModelFile.

Applies to