系统: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)
来源:oschina
链接:https://my.oschina.net/VenusV/blog/4694113