Browse Source

fix a bug in step() of MaskedRNNBase

tags/v0.1.0
xuyige 6 years ago
parent
commit
650838d118
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      fastNLP/modules/encoder/masked_rnn.py

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

@@ -273,7 +273,7 @@ class MaskedRNNBase(nn.Module):
hx = (hx, hx)

func = AutogradMaskedStep(num_layers=self.num_layers,
dropout=self.dropout,
dropout=self.step_dropout,
train=self.training,
lstm=lstm)



Loading…
Cancel
Save