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

Commit

Permalink
Update test_loss.py
Browse files Browse the repository at this point in the history
  • Loading branch information
eric-haibin-lin authored Aug 3, 2018
1 parent eef9d32 commit d2272d6
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions tests/python/unittest/test_loss.py
Original file line number Diff line number Diff line change
Expand Up @@ -143,7 +143,8 @@ def test_kl_loss():
eval_metric=mx.metric.Loss(), optimizer='adam')
assert mod.score(data_iter, eval_metric=mx.metric.Loss())[0][1] < 0.05


# @eric-haibin-lin: Getting rid of fixed seed,
# tracked at https://github.com/apache/incubator-mxnet/issues/11699
@with_seed()
def test_l2_loss():
N = 20
Expand All @@ -161,7 +162,8 @@ def test_l2_loss():
optimizer='adam')
assert mod.score(data_iter, eval_metric=mx.metric.Loss())[0][1] < 0.05


# @eric-haibin-lin: Getting rid of fixed seed,
# tracked at https://github.com/apache/incubator-mxnet/issues/11698
@with_seed()
def test_l1_loss():
N = 20
Expand Down

0 comments on commit d2272d6

Please sign in to comment.