'Returns the current learning rate given the epoch and initial learning rate.
Arguments:
learning_rate (float): Initial learning rate
epoch (int): Current epoch, used to calculate the adjusted learning rate
Returns:
(float): The adjusted learning rate'

def get_learning_rate(self, learning_rate, epoch):

    if (epoch in self.step_config):
        self.steps = self.change[self.step_config.index(epoch)]
    if (self.steps == 0):
        return learning_rate
    else:
        return self.steps
