def METHOD_NAME(self, batch: Batch, batch_idx: int):
    y_hat, y = self.common_step(batch)
    self.val_acc(y_hat.softmax(dim=-1), y)
    self.log('val_acc', self.val_acc, prog_bar=True, on_step=False,
             on_epoch=True)