pspnet-pytorch icon indicating copy to clipboard operation
pspnet-pytorch copied to clipboard

转化onnx是算子不支持,这个有什么办法吗

Open chen-del opened this issue 2 years ago • 5 comments

RuntimeError: Unsupported: ONNX export of operator adaptive_avg_pool2d, since output size is not factor of input size. Please feel free to request support or submit a pull request on PyTorch GitHub.

chen-del avatar Jul 15 '22 09:07 chen-del

你的版本情况是怎么样的

bubbliiiing avatar Jul 17 '22 13:07 bubbliiiing

我也遇到相同的问题

yxxzhuzhuxia avatar Nov 24 '23 12:11 yxxzhuzhuxia

torch 1.10.1+cu111

yxxzhuzhuxia avatar Nov 24 '23 12:11 yxxzhuzhuxia

Starting export with onnx 1.15.0

yxxzhuzhuxia avatar Nov 24 '23 12:11 yxxzhuzhuxia

    torch.onnx.export(self.net,
                    im,
                    f               = model_path,
                    verbose         = False,
                    opset_version   = 12,
                    training        = torch.onnx.TrainingMode.EVAL,
                    do_constant_folding = True,
                    input_names     = input_layer_names,
                    output_names    = output_layer_names,
                    dynamic_axes    = None,
                    operator_export_type=torch.onnx.OperatorExportTypes.ONNX_ATEN_FALLBACK)

yxxzhuzhuxia avatar Nov 28 '23 13:11 yxxzhuzhuxia