RuntimeError: Exporting the operator max_unpool2d to ONNX

…衆ロ難τιáo~ 提交于 2020-10-29 13:35:30

系统:Windows10 CUDA版本:8 9 10.1 10,2 运行平台:Pytorch Torch版本:1.6.0+cu101 报错描述:Pytorch无法导出SegNet到ONNX,不支持max_unpool2d 层,其实是支持的,这锅torch背,极度恶心。

 torch.onnx.export(model, image, "DeepLabv3.onnx",export_params=True, verbose=True,operator_export_type=torch.onnx.OperatorExportTypes.ONNX_ATEN_FALLBACK)

标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!