hummingbird
/
fastNLP
Not watched
2
0
Fork 0
Code
Releases 13 Wiki Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain
Browse Source

update fastNLP/modules/encoder/seq2seq_encoder.py.

tags/v1.0.0alpha
刘向阳 Gitee 3 years ago
parent
e4e6597ac0
commit
9b7562a13a
1 changed files with 1 additions and 1 deletions
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  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 = self.input_fc(x)
x = F.dropout(x, p=self.dropout, training=self.training) 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) encoder_mask = encoder_mask.to(device)


for layer in self.layer_stacks: for layer in self.layer_stacks:


Write Preview
Loading…
Cancel
Save