Unverified Commit 71a5ed34 authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

Fix a `BridgeTower` test (#23694)



fix
Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent 1fe1e3ca
......@@ -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",
]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment