SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Fix a BridgeTower test (#23694)

Browse Source 
fix

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar
2023-05-23 17:32:57 +02:00
committed by GitHub
parent 1fe1e3caa4
commit 71a5ed3433
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/models/bridgetower/test_modeling_bridgetower.py
View File

@@ -627,7 +627,8 @@ class BridgeTowerModelTrainingTest(unittest.TestCase):
non_used_layer_names = ["text_model.pooler"] non_used_layer_names = ["text_model.pooler"]
if model_class == BridgeTowerForMaskedLM: if model_class == BridgeTowerForMaskedLM:
non_used_layer_names = non_used_layer_names + [ 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_image_pooler",
"cross_modal_text_pooler", "cross_modal_text_pooler",
] ]