FX support for ConvNext, Wav2Vec2 and ResNet (#19053)

* Support for ConvNext * Support for Wav2Vec2 * Support for Resnet * Fix small issue in test_modeling_convnext
2022-09-16 10:57:41 +02:00
parent c8e40d6fa1
commit c603c80f46
6 changed files with 117 additions and 3 deletions
--- a/tests/models/convnext/test_modeling_convnext.py
+++ b/tests/models/convnext/test_modeling_convnext.py
@@ -137,6 +137,7 @@ class ConvNextModelTest(ModelTesterMixin, unittest.TestCase):
        else ()
    )

+    fx_compatible = True
    test_pruning = False
    test_resize_embeddings = False
    test_head_masking = False