google-deepmind / google-deepmind/tapnet
OnnxExporterError: Unsupported: ONNX export of operator GridSample with 5D volumetric input.
Open
- 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
Assessment
This issue has not been assessed yet.