add ONNX support for swin transformer (#19390)

* swin transformer onnx support * Updated image dimensions as dynamic Co-authored-by: lewtun <lewis.c.tunstall@gmail.com> Co-authored-by: lewtun <lewis.c.tunstall@gmail.com>
2022-10-07 18:53:24 +05:30
parent 969534af4b
commit e162cebfa3
5 changed files with 30 additions and 2 deletions
--- a/tests/onnx/test_onnx_v2.py
+++ b/tests/onnx/test_onnx_v2.py
@@ -217,6 +217,7 @@ PYTORCH_EXPORT_MODELS = {
    ("longformer", "allenai/longformer-base-4096"),
    ("yolos", "hustvl/yolos-tiny"),
    ("segformer", "nvidia/segformer-b0-finetuned-ade-512-512"),
+    ("swin", "microsoft/swin-tiny-patch4-window7-224"),
 }

 PYTORCH_EXPORT_WITH_PAST_MODELS = {