Fix a BridgeTower test (#23694)

fix Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
2023-05-23 17:32:57 +02:00
parent 1fe1e3caa4
commit 71a5ed3433
1 changed files with 2 additions and 1 deletions
--- a/tests/models/bridgetower/test_modeling_bridgetower.py
+++ b/tests/models/bridgetower/test_modeling_bridgetower.py
@@ -627,7 +627,8 @@ class BridgeTowerModelTrainingTest(unittest.TestCase):
        non_used_layer_names = ["text_model.pooler"]
        if model_class == BridgeTowerForMaskedLM:
            non_used_layer_names = non_used_layer_names + [
-                "cross_modal_image_layers.5",
+                # This number `1` actually depends on the number of layers in `cross_modal_image_layers` (by minus 1)
+                "cross_modal_image_layers.1",
                "cross_modal_image_pooler",
                "cross_modal_text_pooler",
            ]