Add ONNX support for ConvNeXT (#17627)

2022-06-09 15:31:02 +02:00
parent 5323094a22
commit e0be053e43
5 changed files with 34 additions and 2 deletions
--- a/tests/onnx/test_onnx_v2.py
+++ b/tests/onnx/test_onnx_v2.py
@@ -180,6 +180,7 @@ PYTORCH_EXPORT_MODELS = {
    ("ibert", "kssteven/ibert-roberta-base"),
    ("camembert", "camembert-base"),
    ("convbert", "YituTech/conv-bert-base"),
+    ("convnext", "facebook/convnext-tiny-224"),
    ("distilbert", "distilbert-base-cased"),
    ("electra", "google/electra-base-generator"),
    ("resnet", "microsoft/resnet-50"),