Convert to ONNX

by mhrobinson - opened Jun 26, 2022

Jun 26, 2022

I'm trying to convert this to ONNX, but getting an error with the Nested Tensor. Is it possible to convert this to ONNX?

nielsr

Jun 27, 2022

Hi, thanks for your interest in DETR. According to https://github.com/huggingface/transformers/issues/16308, DETR isn't officially supported yet for ONNX export.

cc'ing @lewtun and @regisss here.

Jun 27, 2022

Hi @mhrobinson , I just opened a PR there to easily export DETR to the ONNX format!

Jun 28, 2022

It has been merged!

Jul 11, 2022

Thank you so much!

mhrobinson changed discussion status to closed Jul 11, 2022

Upload images, audio, and videos by dragging in the text input, pasting, or clicking here.

Tap or paste here to upload images

Your need to confirm your account before you can post a new comment.

· Sign up or log in to comment