Add onnx export of models with a multiple choice classification head (#16758)

* Add export of models with a multiple-choice classification head
2022-04-19 15:51:51 +02:00
parent b74a955325
commit 77de8d6c31
19 changed files with 134 additions and 41 deletions
--- a/tests/onnx/test_onnx_v2.py
+++ b/tests/onnx/test_onnx_v2.py
@@ -182,6 +182,7 @@ PYTORCH_EXPORT_MODELS = {
    ("layoutlm", "microsoft/layoutlm-base-uncased"),
    ("vit", "google/vit-base-patch16-224"),
    ("beit", "microsoft/beit-base-patch16-224"),
+    ("data2vec-text", "facebook/data2vec-text-base"),
 }

 PYTORCH_EXPORT_WITH_PAST_MODELS = {