def test_learning_rate_value(self): lr = -1.0 with pytest.raises(ValueError): lr_schedules.ExponentialDecayLR(lr, decay_rate, decay_steps) with pytest.raises(ValueError): lr_schedules.PolynomialDecayLR(lr, end_learning_rate, decay_steps, power)
def test_learning_rate_type(self): lr = True with pytest.raises(TypeError): lr_schedules.ExponentialDecayLR(lr, decay_rate, decay_steps) with pytest.raises(TypeError): lr_schedules.PolynomialDecayLR(lr, end_learning_rate, decay_steps, power)
def test_decay_steps_value(self): decay_steps_e = -2 with pytest.raises(ValueError): lr_schedules.ExponentialDecayLR(learning_rate, decay_rate, decay_steps_e) with pytest.raises(ValueError): lr_schedules.CosineDecayLR(min_lr, max_lr, decay_steps_e) with pytest.raises(ValueError): lr_schedules.PolynomialDecayLR(learning_rate, end_learning_rate, decay_steps_e, power)
def test_decay_rate_value(self): rate = -1.0 with pytest.raises(ValueError): lr_schedules.ExponentialDecayLR(learning_rate, rate, decay_steps)
def test_decay_rate_type(self): rate = 'a' with pytest.raises(TypeError): lr_schedules.ExponentialDecayLR(learning_rate, rate, decay_steps)
def test_exponential_decay(): lr_schedule = lr_schedules.ExponentialDecayLR(learning_rate, decay_rate, decay_steps, True) _executor.compile(lr_schedule, global_step)
def test_is_stair(self): is_stair = 1 with pytest.raises(TypeError): lr_schedules.ExponentialDecayLR(learning_rate, decay_rate, decay_steps, is_stair)