diff --git a/tests/python/unittest/test_gluon_rnn.py b/tests/python/unittest/test_gluon_rnn.py index b410362c8fd1..c2f5ebbdad90 100644 --- a/tests/python/unittest/test_gluon_rnn.py +++ b/tests/python/unittest/test_gluon_rnn.py @@ -23,7 +23,7 @@ import unittest from mxnet.test_utils import almost_equal, assert_almost_equal from common import assert_raises_cudnn_not_satisfied - +from tests.python.unittest.common import with_seed def test_rnn(): cell = gluon.rnn.RNNCell(100, prefix='rnn_') @@ -244,6 +244,7 @@ def test_bidirectional(): @assert_raises_cudnn_not_satisfied(min_version='5.1.10') +@with_seed() def test_layer_bidirectional(): class RefBiLSTM(gluon.Block): def __init__(self, size, **kwargs): @@ -279,7 +280,7 @@ def forward(self, inpt): ref_net_params[k.replace('l0', 'l0l0').replace('r0', 'r0l0')].set_data(weights[k]) data = mx.random.uniform(shape=(11, 10, in_size)) - assert_allclose(net(data).asnumpy(), ref_net(data).asnumpy()) + assert_allclose(net(data).asnumpy(), ref_net(data).asnumpy(), rtol=1e-04, atol=1e-02)