GH-61: Remove flaten parameters

flairNLP · Oct 4, 2018 · 677f727 · 677f727
1 parent bdd34f9
commit 677f727
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 4 deletions.
diff --git a/flair/embeddings.py b/flair/embeddings.py
@@ -669,8 +669,6 @@ def embed(self, sentences: Union[List[Sentence], Sentence]):
 
         packed = torch.nn.utils.rnn.pack_padded_sequence(sentence_tensor, lengths)
 
-        self.rnn.flatten_parameters()
-
         lstm_out, hidden = self.rnn(packed)
 
         outputs, output_lengths = torch.nn.utils.rnn.pad_packed_sequence(lstm_out)

diff --git a/flair/models/language_model.py b/flair/models/language_model.py
@@ -66,8 +66,6 @@ def forward(self, input, hidden, ordered_sequence_lengths=None):
         encoded = self.encoder(input)
         emb = self.drop(encoded)
 
-        self.rnn.flatten_parameters()
-
         output, hidden = self.rnn(emb, hidden)
 
         if self.proj is not None: