pspnet-pytorch
pspnet-pytorch copied to clipboard
转化onnx是算子不支持,这个有什么办法吗
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.
你的版本情况是怎么样的
我也遇到相同的问题
torch 1.10.1+cu111
Starting export with onnx 1.15.0
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)