onnx的pytorch模型部署方法
torch模型转onnx
在model.eval()
和output = modle(input)
后面添加一句话:
torch.onnx.export(model, input, 'modelname.onnx')
即可导出
- onnx支持的算子,可以在 算子文档 中查看
- 如果不支持报错,怎么查看具体是哪个算子不支持呢?我也不知道。
- 在 PyTorch 中支持更多 ONNX 算子
onnx模型的使用
不知。。
有空看看部署在安卓设备的方法。(我们新生项目小组已经做好了,用的是maui+onnx )
(105条消息) 在Android设备部署PyTorch模型_紫芝的博客-CSDN博客_android pytorch
(105条消息) 将Pytorch模型部署到Android端_YDDUONG的博客-CSDN博客_pytorch模型部署到安卓
maui
electron
onnx模型的优点
不知。。
多想多做,发篇一作