def forward_loss(self, sentences: Union[(List[DataPoint], DataPoint)]) -> torch.tensor:
    (scores, labels) = self.forward_pass(sentences)
    return self._calculate_loss(scores, labels)
def forward_loss(self, sentences: Union[(List[DataPoint], DataPoint)]) -> torch.tensor:
    (scores, labels) = self.forward_pass(sentences)
    return self._calculate_loss(scores, labels)