From d2272d6269221332a672068b0fb397fb437a6397 Mon Sep 17 00:00:00 2001 From: Haibin Lin Date: Fri, 3 Aug 2018 09:35:05 -0700 Subject: [PATCH] Update test_loss.py --- tests/python/unittest/test_loss.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/tests/python/unittest/test_loss.py b/tests/python/unittest/test_loss.py index fea8e5f26693..887cba3cd9ed 100644 --- a/tests/python/unittest/test_loss.py +++ b/tests/python/unittest/test_loss.py @@ -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 @@ -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