def METHOD_NAME(self, val_batch, batch_idx):
    """
    Performs validation of data in batches
    :param val_batch: Batch data
    :param batch_idx: Batch indices
    :return: output - valid step loss
    """
    x, y = val_batch
    logits = self.forward(x)
    loss = self.cross_entropy_loss(logits, y)
    self.val_outputs.append(loss)
    return {"val_loss": loss}