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