def __init__(self, min_lr=0.0, max_lr=0.05, base_epochs=10, mul_epochs=2):
    super(SGDR, self).__init__()
    self.min_lr = min_lr
    self.max_lr = max_lr
    self.base_epochs = base_epochs
    self.mul_epochs = mul_epochs
    self.cycles = 0.0
    self.cycle_iterations = 0.0
    self.trn_iterations = 0.0
    self._reset()
def __init__(self, min_lr=0.0, max_lr=0.05, base_epochs=10, mul_epochs=2):
    super(SGDR, self).__init__()
    self.min_lr = min_lr
    self.max_lr = max_lr
    self.base_epochs = base_epochs
    self.mul_epochs = mul_epochs
    self.cycles = 0.0
    self.cycle_iterations = 0.0
    self.trn_iterations = 0.0
    self._reset()