def METHOD_NAME(self, stage, stage_loss, epoch=None):
    "Gets called when a stage (either training, validation, test) starts."
    if stage == sb.Stage.TRAIN:
        self.train_loss = stage_loss
    if stage == sb.Stage.VALID:
        print("Epoch %d complete" % epoch)
        print("Train loss: %.2f" % self.train_loss)
        print("Valid loss: %.2f" % stage_loss)
        print("Recalculating and recording alignments...")
        self.evaluate(self.hparams.train_data)