def METHOD_NAME(self, batch, stage):
    """Forward computations from the sentence batches to the output probabilities."""
    batch = batch.to(self.device)
    tokens_bos, _ = batch.tokens_bos
    logits = self.hparams.model(tokens_bos)
    pred = self.hparams.log_softmax(logits)
    return pred