RuntimeError: Exporting the operator max_unpool2d to ONNX

系统: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)

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章