From 88c50d96a52d6993341e53c3f24528c0bee2f5a0 Mon Sep 17 00:00:00 2001 From: ydshieh Date: Tue, 23 May 2023 17:04:29 +0200 Subject: [PATCH] fix --- tests/models/bridgetower/test_modeling_bridgetower.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/models/bridgetower/test_modeling_bridgetower.py b/tests/models/bridgetower/test_modeling_bridgetower.py index 1a4d34d3967f98..9c40f376a7b573 100644 --- a/tests/models/bridgetower/test_modeling_bridgetower.py +++ b/tests/models/bridgetower/test_modeling_bridgetower.py @@ -627,7 +627,8 @@ def _get_non_used_layer_names(self, model_class): 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", ]