GH-61: Revert last change.

flairNLP · Oct 4, 2018 · 05a68fd · 05a68fd
1 parent 5ca09ff
commit 05a68fd
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 1 deletion.
diff --git a/flair/embeddings.py b/flair/embeddings.py
@@ -669,6 +669,8 @@ 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,6 +66,8 @@ 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:

diff --git a/requirements.txt b/requirements.txt
@@ -9,4 +9,3 @@ matplotlib==3.0.0
 mpld3==0.3
 jinja2==2.10
 sklearn==0.0
-google-compute-engine==2.8.3