Add OnnxConfig for ConvBERT (#16859)

* add OnnxConfig for ConvBert Co-authored-by: ChainYo <t.chaigneau.tc@gmail.com>
2022-04-22 18:19:15 +02:00
parent 0d1cff1195
commit ec81c11a18
5 changed files with 35 additions and 2 deletions
--- a/tests/onnx/test_onnx_v2.py
+++ b/tests/onnx/test_onnx_v2.py
@@ -175,6 +175,7 @@ PYTORCH_EXPORT_MODELS = {
    ("bigbird", "google/bigbird-roberta-base"),
    ("ibert", "kssteven/ibert-roberta-base"),
    ("camembert", "camembert-base"),
+    ("convbert", "YituTech/conv-bert-base"),
    ("distilbert", "distilbert-base-cased"),
    ("electra", "google/electra-base-generator"),
    ("roberta", "roberta-base"),