Add ONNX support for DETR (#17904)

2022-06-28 14:48:43 +02:00
parent bfcd5743ee
commit 76d13de5ae
6 changed files with 55 additions and 2 deletions
--- a/tests/onnx/test_onnx_v2.py
+++ b/tests/onnx/test_onnx_v2.py
@@ -183,6 +183,7 @@ PYTORCH_EXPORT_MODELS = {
    ("deberta", "microsoft/deberta-base"),
    ("deberta-v2", "microsoft/deberta-v2-xlarge"),
    ("convnext", "facebook/convnext-tiny-224"),
+    ("detr", "facebook/detr-resnet-50"),
    ("distilbert", "distilbert-base-cased"),
    ("electra", "google/electra-base-generator"),
    ("resnet", "microsoft/resnet-50"),