From 52397d92de0a0dad599abd8f01e361785d96a255 Mon Sep 17 00:00:00 2001 From: Rahul Date: Fri, 17 Aug 2018 15:08:16 -0700 Subject: [PATCH] remove debugging assert --- tests/python/unittest/test_optimizer.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/tests/python/unittest/test_optimizer.py b/tests/python/unittest/test_optimizer.py index 06271923eaed..e79b9a1d0f28 100644 --- a/tests/python/unittest/test_optimizer.py +++ b/tests/python/unittest/test_optimizer.py @@ -1068,7 +1068,6 @@ def test_poly_scheduler(): steps = 1000 poly_sched = mx.lr_scheduler.PolyScheduler(steps, base_lr=base_lr, pwr=2, final_lr=final_lr, warmup_steps=100, warmup_begin_lr=0, warmup_mode='linear') - np.testing.assert_almost_equal(poly_sched(0), 0) np.testing.assert_almost_equal(poly_sched(50), float(base_lr)/2) np.testing.assert_almost_equal(poly_sched(100), base_lr) @@ -1085,7 +1084,6 @@ def test_cosine_scheduler(): np.testing.assert_almost_equal(cosine_sched(0), base_lr) np.testing.assert_almost_equal(cosine_sched(steps), final_lr) assert (cosine_sched(500) > 1.5) - assert False if __name__ == '__main__': import nose