def METHOD_NAME(self, epoch):
    """Update the learning rate at the beginning of the given epoch."""
    self.lr = self.get_next_lr(epoch)
    self.optimizer.set_lr(self.lr)
    return self.optimizer.get_lr()