Wrong shape from average pool about onnx2torch

[JohnMasoner]

I was able to reproduce it, the problem is in the PyTorch tracing mechanism. The fastest and simplest solution is to update your PyTorch to 1.13.x (if it possible for you).

Many Thanks @senysenyseny16,

I updated the PyTorch to 1.13.1 to solve this problem.

[JohnMasoner]

I temporarily set spatial rank to 2 and solved my problem. But this is obviously a temporary solution and I am looking for options to solve it.

[JohnMasoner]

I've just looked at #121 and this seems to be the same issue.

[senysenyseny16]

Hi @JohnMasoner

Could you check conversion with onnx2torch==1.5.6?

pip install onnx2torch==1.5.6

[JohnMasoner]

Hi @JohnMasoner

Could you check conversion with onnx2torch==1.5.6?

pip install onnx2torch==1.5.6

Thank you @senysenyseny16
I still got the same bug.

[senysenyseny16]

I was able to reproduce it, the problem is in the PyTorch tracing mechanism.
The fastest and simplest solution is to update your PyTorch to 1.13.x (if it possible for you).

[senysenyseny16]

Also, I think the torch.export in PyTorch 1.12.x slightly breaks the model:

❯ python3 /venv39/lib/python3.9/site-packages/onnxruntime/tools/symbolic_shape_infer.py --input monai_1_12.onnx --output monai_1_12_infer.onnx
Traceback (most recent call last):
  File ".../symbolic_shape_infer.py", line 2411, in <module>
    out_mp = SymbolicShapeInference.infer_shapes(
  File ".../symbolic_shape_infer.py", line 2347, in infer_shapes
    raise Exception("Incomplete symbolic shape inference")
Exception: Incomplete symbolic shape inference

On PyTorch 1.13.x shape inference works as expected.

[tan-may16]

Hello, I am facing a similar issue. But following the above discussion is not working. This is the error I am getting:
NotImplementedError: BatchNorm operation with spatial rank == -2 is not implemented

please let me know what could be done to resolve the issue. Thank you.

[RainFrost1]

Hello, I am facing a similar issue. But following the above discussion is not working. This is the error I am getting: NotImplementedError: BatchNorm operation with spatial rank == -2 is not implemented

please let me know what could be done to resolve the issue. Thank you.

Have you find a solution?