Merge branch 'master' into gh-3487/fix-gpu-memory-leak-text-pair-regr…

…essor
flairNLP · Jul 8, 2024 · 27c0638 · 27c0638
2 parents d0842db + 355e2c0
commit 27c0638
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/flair/models/text_regression_model.py b/flair/models/text_regression_model.py
@@ -64,7 +64,8 @@ def forward_loss(self, sentences: List[Sentence]) -> Tuple[torch.Tensor, int]:
 
     def _labels_to_tensor(self, sentences: List[Sentence]):
         indices = [
-            torch.tensor([float(label.value) for label in sentence.labels], dtype=torch.float) for sentence in sentences
+            torch.tensor([float(label.value) for label in sentence.get_labels(self.label_name)], dtype=torch.float)
+            for sentence in sentences
         ]
 
         vec = torch.cat(indices, 0).to(flair.device)