onnxruntime - экспорт scikit RandomForestClassifier не выполняется при импорте

Я экспортировал модель scikit RandomForestClassifier в ONNX, используя WinMLTools. Когда я пытался импортировать файл, используя ONNXRuntime (на C#). Я получаю следующую ошибку:

System.AccessViolationException: 'Attempted to read or write protected memory. This is often an indication that other memory is corrupt.'

Это происходит после того, как второе имя выхода (="вероятности") передано:

NativeMethods.OrtCastTypeInfoToTensorInfo(typeInfo)

и это приводит к IntPtr адреса 0x000000000.

Входной параметр и 1-й выходной параметр работают, как запланировано.

Вывод модели представляет собой массив чисел с плавающей точкой, поэтому я не понимаю, в чем может быть проблема.

Полный стек вызовов:

   at Microsoft.ML.OnnxRuntime.NativeMethods.OrtGetTensorElementType(IntPtr typeAndShapeInfo)
   at Microsoft.ML.OnnxRuntime.InferenceSession.GetMetadataFromTypeInfo(IntPtr typeInfo) in C:\agent\_work\7\s\csharp\src\Microsoft.ML.OnnxRuntime\InferenceSession.cs:line 305

0 ответов

Другие вопросы по тегам