Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Commit

Permalink
remove debugging assert
Browse files Browse the repository at this point in the history
  • Loading branch information
rahul003 committed Aug 17, 2018
1 parent 44a5af4 commit 52397d9
Showing 1 changed file with 0 additions and 2 deletions.
2 changes: 0 additions & 2 deletions tests/python/unittest/test_optimizer.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand All @@ -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
Expand Down

0 comments on commit 52397d9

Please sign in to comment.