Merge pull request #154 from zalandoresearch/GH-149-predict-clear-emb…

…eddings

GH-149: Always clear embeddings in predict

flair/models/text_classification_model.py

@@ -99,7 +99,7 @@ def load_from_file(cls, model_file):
         model.eval()
         return model
 
-    def predict(self, sentences: Union[Sentence, List[Sentence]], mini_batch_size: int = 32, embeddings_in_memory: bool = True) -> List[Sentence]:
+    def predict(self, sentences: Union[Sentence, List[Sentence]], mini_batch_size: int = 32) -> List[Sentence]:
         """
         Predicts the class labels for the given sentences. The labels are directly added to the sentences.
         :param sentences: list of sentences
@@ -118,8 +118,7 @@ def predict(self, sentences: Union[Sentence, List[Sentence]], mini_batch_size: i
             for (sentence, labels) in zip(batch, predicted_labels):
                 sentence.labels = labels
 
-            if not embeddings_in_memory:
-                clear_embeddings(batch)
+            clear_embeddings(batch)
 
         return sentences