Поделиться через


OnnxOptions Класс

Определение

Параметры для OnnxScoringEstimator.

public sealed class OnnxOptions
type OnnxOptions = class
Public NotInheritable Class OnnxOptions
Наследование
OnnxOptions

Конструкторы

OnnxOptions()

Параметры для OnnxScoringEstimator.

Поля

FallbackToCpu

Если значение равно true, возобновляет выполнение на ЦП при ошибке GPU. Если задано значение false, вызовет исключение GPU.

GpuDeviceId

Идентификатор устройства GPU для запуска (например, 0,1,..). Значение NULL для ЦП. Требуется CUDA 10.1.

InputColumns

Имя входного столбца.

InterOpNumThreads

Управляет количеством потоков, используемых для параллелизации выполнения графа (между узлами).

IntraOpNumThreads

Управляет количеством потоков, используемых для запуска модели.

ModelFile

Путь к файлу модели onnx.

OutputColumns

Имя выходного столбца.

RecursionLimit

Предел рекурсии Protobuf CodedInputStream.

ShapeDictionary

Фигуры ONNX, которые будут использоваться по сравнению с фигурами, загруженными из ModelFile.

Применяется к