def METHOD_NAME(self, epoch, val_loss=None):
    """Update the learning rate at the end of the given epoch."""
    super().METHOD_NAME(epoch, val_loss)
    # we don't change the learning rate at epoch boundaries
    return self.optimizer.get_lr()