Exemplo n.º 1
0
def test_step_scheduler(init_lr, max_iter, gamma, iter_steps):
    scheduler_tester(lrs.StepScheduler, RefStep, max_iter,
                     [init_lr, gamma, iter_steps])
Exemplo n.º 2
0
def test_cosine_scheduler(init_lr, max_iter):
    scheduler_tester(lrs.CosineScheduler, RefCosine, max_iter,
                     [init_lr, max_iter])
Exemplo n.º 3
0
def test_polynomial_scheduler(init_lr, max_iter, power):
    scheduler_tester(lrs.PolynomialScheduler, RefPolynomial, max_iter,
                     [init_lr, max_iter, power])
Exemplo n.º 4
0
def test_exponential_scheduler(init_lr, max_iter, gamma, iter_interval):
    scheduler_tester(lrs.ExponentialScheduler, RefExponential, max_iter,
                     [init_lr, gamma, iter_interval])