!4 update fastNLP/modules/encoder/seq2seq_encoder.py.

Merge pull request !4 from 刘向阳/N/A
NLP-fork · Nov 18, 2020 · c49c849 · c49c849
2 parents e4e6597 + 9b7562a
commit c49c849
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/fastNLP/modules/encoder/seq2seq_encoder.py b/fastNLP/modules/encoder/seq2seq_encoder.py
@@ -132,7 +132,7 @@ def forward(self, tokens, seq_len):
         x = self.input_fc(x)
         x = F.dropout(x, p=self.dropout, training=self.training)
 
-        encoder_mask = seq_len_to_mask(seq_len)
+        encoder_mask = seq_len_to_mask(seq_len, max_len=max_src_len)
         encoder_mask = encoder_mask.to(device)
 
         for layer in self.layer_stacks: