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)

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