嗨,加布里埃尔,
ONNX导出功能的新版本即将发布。新版本的LSTM支万博1manbetx持将会更好。
ONNX导出在Windows ML中产生错误
4次浏览(最近30天)
显示旧的评论
我想用ONNX导出函数导出我的深度神经网络，并在Windows ML中使用它
WinMLRunner
，它只是检查.onnx文件。这似乎适用于cnn网络，但是对于lstm网络，我收到的错误是:“第一次输入没有排名2”。如果我在网络中有不止一个lstm层，错误消息就会以某种方式变成:“第一个输入张量必须有秩3”。
我附加了一个带有以下简单LSTM-network的文件和另一个带有相应ONNX导出的文件(Windows ML需要onset v7或v8):
层= [．．．
sequenceInputLayer (14)
lstmLayer (100“OutputMode”，“最后一次”）;
fullyConnectedLayer (1)
regressionLayer];
我导出的网络如下:
exportONNXNetwork (lstmNet“lstm_net_onset_v7.onnx”，“OpsetVersion”7)
然后用WinMLRunner测试它:
WinMLRunner.exe模型C: \ lstm_net_onset_v7。onnx cpu
感谢您对如何解决这个问题提出的任何建议。我已经在ONNX转换器的文件交换页面上发布了这个问题。
0评论
更多的答案(0)
另请参阅
类别
翻译的
你也可以从以下列表中选择一个网站: