Add Onnx Config for PoolFormer (#20868)

poolformer onnx Co-authored-by: syed <syed.abdul@sandlogic.com>
2022-12-23 12:00:57 +05:30
parent 4a4cd6cd02
commit 15bc776fec
5 changed files with 39 additions and 2 deletions
--- a/tests/onnx/test_onnx_v2.py
+++ b/tests/onnx/test_onnx_v2.py
@@ -210,6 +210,7 @@ PYTORCH_EXPORT_MODELS = {
    ("owlvit", "google/owlvit-base-patch32"),
    ("perceiver", "hf-internal-testing/tiny-random-PerceiverModel", ("masked-lm", "sequence-classification")),
    ("perceiver", "hf-internal-testing/tiny-random-PerceiverModel", ("image-classification",)),
+    ("poolformer", "sail/poolformer_s12"),
    ("rembert", "google/rembert"),
    ("resnet", "microsoft/resnet-50"),
    ("roberta", "hf-internal-testing/tiny-random-RobertaModel"),