def test_step_scheduler(init_lr, max_iter, gamma, iter_steps):
scheduler_tester(lrs.StepScheduler, RefStep, max_iter,
[init_lr, gamma, iter_steps])
def test_cosine_scheduler(init_lr, max_iter):
scheduler_tester(lrs.CosineScheduler, RefCosine, max_iter,
[init_lr, max_iter])
def test_polynomial_scheduler(init_lr, max_iter, power):
scheduler_tester(lrs.PolynomialScheduler, RefPolynomial, max_iter,
[init_lr, max_iter, power])
def test_exponential_scheduler(init_lr, max_iter, gamma, iter_interval):
scheduler_tester(lrs.ExponentialScheduler, RefExponential, max_iter,
[init_lr, gamma, iter_interval])