Skip to content

Commit 3eefc2c

Browse files
committed
Update modeling_lxmert.py
1 parent f7d5963 commit 3eefc2c

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

src/transformers/models/lxmert/modeling_lxmert.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1074,6 +1074,9 @@ def __init__(self, config):
10741074
}
10751075
self.visual_losses = visual_losses
10761076

1077+
def _tie_weights(self):
1078+
self.cls.predictions.decoder.weight = self.lxmert.embeddings.word_embeddings.weight
1079+
10771080
def resize_token_embeddings(
10781081
self, new_num_tokens: int, pad_to_multiple_of: Optional[int] = None, mean_resizing: bool = True
10791082
) -> nn.Embedding:

0 commit comments

Comments
 (0)