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/resnet/test_modeling_resnet.py
+++ b/tests/models/resnet/test_modeling_resnet.py
@@ -126,6 +126,7 @@ class ResNetModelTest(ModelTesterMixin, unittest.TestCase):

    all_model_classes = (ResNetModel, ResNetForImageClassification) if is_torch_available() else ()

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