google-deepmind / google-deepmind/tapnet

OnnxExporterError: Unsupported: ONNX export of operator GridSample with 5D volumetric input.

Open
#79 10 comments 1 reaction 0 assignees View on GitHub
Dominant language
Jupyter Notebook
Stars
2k
Forks
192
PR merge metrics
No merged PRs in 30d

Description

Hi everyone,

Thanks for the awesome work. I've been trying to export the pytorch model to ONNX for inference with `torch.onnx.export` but it yields this error : `OnnxExporterError: Unsupported: ONNX export of operator GridSample with 5D volumetric input.`

Unfortunately, It seems 5D grid_sample is still unsupported by onnx / torch. Is there any alternative available ? Or any advice to make the model work with ONNX ?

Thanks

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.