- Supporto ONNX
- Conversione Di Formato
- Conversione automatica del tipo di dati
- Creazione di un Modello
- Esecuzione di un modello
- Eseguire nello Strategy Tester
- OnnxCreate
- OnnxCreateFromBuffer
- OnnxRelease
- OnnxRun
- OnnxGetInputCount
- OnnxGetOutputCount
- OnnxGetInputName
- OnnxGetOutputName
- OnnxGetInputTypeInfo
- OnnxGetOutputTypeInfo
- OnnxSetInputShape
- OnnxSetOutputShape
- Strutture dati
OnnxRun
Esegue un modello ONNX.
bool OnnxRun(
|
Parametri
onnx_handle
[in] Handle ONNX dell'oggetto della sessione creato tramite OnnxCreate o OnnxCreateFromBuffer.
flags
[in] Flag da ENUM_ONNX_FLAGS che descrive la modalità di esecuzione: ONNX_DEBUG_LOGS e ONNX_NO_CONVERSION.
...
[in] [out] Ingressi e uscite del modello.
Restituisce true in caso di successo o false altrimenti. Per ottenere il codice errore, chiamare la funzione GetLastError.
ENUM_ONNX_FLAGS
ID |
Descrizione |
|---|---|
ONNX_LOGLEVEL_VERBOSE |
Registra tutti i messaggi |
ONNX_LOGLEVEL_INFO |
Registra messaggi informativi, avvisi ed errori (questo flag sostituisce ONNX_DEBUG_LOGS) |
ONNX_LOGLEVEL_WARNING |
Registra avvisi ed errori (predefinito) |
ONNX_LOGLEVEL_ERROR |
Registra solo gli errori |
ONNX_NO_CONVERSION |
Disabilita la conversione automatica, utilizza i dati utente così come sono |
ONNX_COMMON_FOLDER |
Carica un file modello dalla cartella Common\Files; il valore è uguale al flag FILE_COMMON |
ONNX_USE_CPU_ONLY |
Esegue il modello ONNX utilizzando solo la CPU |
ONNX_GPU_DEVICE_0 |
Dispositivo CUDA con indice 0 (predefinito) |
ONNX_GPU_DEVICE_1 |
Dispositivo CUDA con indice 1 * |
ONNX_GPU_DEVICE_2 |
Dispositivo CUDA con indice 2 * |
ONNX_GPU_DEVICE_3 |
Dispositivo CUDA con indice 3 * |
ONNX_GPU_DEVICE_4 |
Dispositivo CUDA con indice 4 * |
ONNX_GPU_DEVICE_5 |
Dispositivo CUDA con indice 5 * |
ONNX_GPU_DEVICE_6 |
Dispositivo CUDA con indice 6 * |
ONNX_GPU_DEVICE_7 |
Dispositivo CUDA con indice 7 * |
ONNX_ENABLE_PROFILING |
Abilita la profilazione del modello ONNX |
* I flag del formato ONNX_GPU_DEVICE_N devono essere utilizzati sui sistemi con due o più GPU compatibili con CUDA. Se vengono specificati più flag di selezione GPU, verrà utilizzato il dispositivo con l'indice più basso.
Se viene specificato un indice del dispositivo inesistente, la GPU verrà selezionata automaticamente.
Esempio:
const long ExtOutputShape[] = {1,1}; // dimensione del modello d'uscita
|
Vedi anche