OnnxOptions Classe

Definição

As opções para um OnnxScoringEstimator.

public sealed class OnnxOptions
type OnnxOptions = class
Public NotInheritable Class OnnxOptions
Herança
OnnxOptions

Construtores

OnnxOptions()

As opções para um OnnxScoringEstimator.

Campos

FallbackToCpu

Se for true, retomará a execução na CPU após um erro de GPU. Se false, gerará a exceção de GPU.

GpuDeviceId

ID do dispositivo GPU em que executar (por exemplo, 0,1,..). Nulo para CPU. Requer CUDA 10.1.

InputColumns

Nome da coluna de entrada.

InterOpNumThreads

Controla o número de threads usados para paralelizar a execução do grafo (entre nós).

IntraOpNumThreads

Controla o número de threads a serem usados para executar o modelo.

ModelFile

Caminho para o arquivo de modelo onnx.

OutputColumns

Nome da coluna de saída.

RecursionLimit

Limite de recursão de CodedInputStream do Protobuf.

ShapeDictionary

Formas ONNX a serem usadas sobre aquelas carregadas de ModelFile.

Aplica-se a