Fix Bert dense

2023-03-05 19:24:14 -08:00 · 2023-03-05 19:24:14 -08:00 · a788e67ef2
commit a788e67ef2
parent 32cb51ae38
1 changed files with 0 additions and 1 deletions
--- a/examples/fairseq/models/bert.py
+++ b/examples/fairseq/models/bert.py
@ -375,7 +375,6 @@ class BertModel(BaseFairseqModel):
            src_tokens, features_only=True, return_all_hiddens=return_all_hiddens
        )
        x, extra = encoder_out["encoder_out"], encoder_out
-        x = x.transpose(0, 1)

        if classification_head_name is not None:
            x = self.classification_heads[classification_head_name](x)