Browse Source

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

Merge pull request !4 from 刘向阳/N/A
tags/v1.0.0alpha
WillQvQ Gitee 3 years ago
parent
commit
c49c849aee
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      fastNLP/modules/encoder/seq2seq_encoder.py

+ 1
- 1
fastNLP/modules/encoder/seq2seq_encoder.py View File

@@ -132,7 +132,7 @@ class TransformerSeq2SeqEncoder(Seq2SeqEncoder):
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:


Loading…
Cancel
Save