提交 95d3d9cb authored 作者: Frederic Bastien's avatar Frederic Bastien

Add seeding of tests to help investigate why it fail in jenkins

上级 8035c263
...@@ -102,6 +102,8 @@ def test_dnn_conv_inplace(): ...@@ -102,6 +102,8 @@ def test_dnn_conv_inplace():
""" """
if not dnn.dnn_available(test_ctx_name): if not dnn.dnn_available(test_ctx_name):
raise SkipTest(dnn.dnn_available.msg) raise SkipTest(dnn.dnn_available.msg)
utt.seed_rng()
img_shp = [2, 5, 6, 8] img_shp = [2, 5, 6, 8]
kern_shp = [3, 5, 5, 6] kern_shp = [3, 5, 5, 6]
img = T.tensor4('img') img = T.tensor4('img')
...@@ -150,6 +152,7 @@ def test_dnn_conv_inplace(): ...@@ -150,6 +152,7 @@ def test_dnn_conv_inplace():
def test_pooling(): def test_pooling():
if not dnn.dnn_available(test_ctx_name): if not dnn.dnn_available(test_ctx_name):
raise SkipTest(dnn.dnn_available.msg) raise SkipTest(dnn.dnn_available.msg)
utt.seed_rng()
# 'average_exc_pad' is disabled for versions < 4004 # 'average_exc_pad' is disabled for versions < 4004
if dnn.version(raises=False) < 4004: if dnn.version(raises=False) < 4004:
...@@ -241,6 +244,8 @@ def test_pooling(): ...@@ -241,6 +244,8 @@ def test_pooling():
def test_pooling_with_tensor_vars(): def test_pooling_with_tensor_vars():
if not dnn.dnn_available(test_ctx_name): if not dnn.dnn_available(test_ctx_name):
raise SkipTest(dnn.dnn_available.msg) raise SkipTest(dnn.dnn_available.msg)
utt.seed_rng()
x = T.tensor4() x = T.tensor4()
ws = theano.shared(np.array([2, 2], dtype='int32')) ws = theano.shared(np.array([2, 2], dtype='int32'))
stride = theano.shared(np.array([1, 1], dtype='int32')) stride = theano.shared(np.array([1, 1], dtype='int32'))
...@@ -295,6 +300,7 @@ def test_pooling3d(): ...@@ -295,6 +300,7 @@ def test_pooling3d():
# 3d pooling requires version 3 or newer. # 3d pooling requires version 3 or newer.
if not dnn.dnn_available(test_ctx_name) or dnn.version(raises=False) < 3000: if not dnn.dnn_available(test_ctx_name) or dnn.version(raises=False) < 3000:
raise SkipTest(dnn.dnn_available.msg) raise SkipTest(dnn.dnn_available.msg)
utt.seed_rng()
# We force the FAST_RUN as we don't want the reference to run in DebugMode. # We force the FAST_RUN as we don't want the reference to run in DebugMode.
mode_without_gpu_ref = theano.compile.mode.get_mode( mode_without_gpu_ref = theano.compile.mode.get_mode(
...@@ -383,6 +389,7 @@ def test_pooling3d(): ...@@ -383,6 +389,7 @@ def test_pooling3d():
def test_pooling_opt(): def test_pooling_opt():
if not dnn.dnn_available(test_ctx_name): if not dnn.dnn_available(test_ctx_name):
raise SkipTest(dnn.dnn_available.msg) raise SkipTest(dnn.dnn_available.msg)
utt.seed_rng()
# 2D pooling # 2D pooling
x = T.matrix() x = T.matrix()
...@@ -457,6 +464,7 @@ def test_pooling_opt_arbitrary_dimensions(): ...@@ -457,6 +464,7 @@ def test_pooling_opt_arbitrary_dimensions():
if not dnn.dnn_available(test_ctx_name): if not dnn.dnn_available(test_ctx_name):
raise SkipTest(dnn.dnn_available.msg) raise SkipTest(dnn.dnn_available.msg)
utt.seed_rng()
# 'average_exc_pad' is disabled for versions < 4004 # 'average_exc_pad' is disabled for versions < 4004
if dnn.version(raises=False) < 4004: if dnn.version(raises=False) < 4004:
...@@ -875,6 +883,8 @@ def test_dnn_conv_border_mode(): ...@@ -875,6 +883,8 @@ def test_dnn_conv_border_mode():
def test_dnn_conv_alpha_output_merge(): def test_dnn_conv_alpha_output_merge():
if not dnn.dnn_available(test_ctx_name): if not dnn.dnn_available(test_ctx_name):
raise SkipTest(dnn.dnn_available.msg) raise SkipTest(dnn.dnn_available.msg)
utt.seed_rng()
img = T.tensor4() img = T.tensor4()
kern = T.tensor4() kern = T.tensor4()
out = T.tensor4() out = T.tensor4()
...@@ -938,6 +948,8 @@ def test_dnn_conv_alpha_output_merge(): ...@@ -938,6 +948,8 @@ def test_dnn_conv_alpha_output_merge():
def test_dnn_conv_grad(): def test_dnn_conv_grad():
if not dnn.dnn_available(test_ctx_name): if not dnn.dnn_available(test_ctx_name):
raise SkipTest(dnn.dnn_available.msg) raise SkipTest(dnn.dnn_available.msg)
utt.seed_rng()
b = 1 b = 1
c = 4 c = 4
f = 3 f = 3
...@@ -1010,6 +1022,7 @@ def test_conv3d_fwd(): ...@@ -1010,6 +1022,7 @@ def test_conv3d_fwd():
if not dnn.dnn_available(test_ctx_name): if not dnn.dnn_available(test_ctx_name):
raise SkipTest(dnn.dnn_available.msg) raise SkipTest(dnn.dnn_available.msg)
utt.seed_rng()
def run_conv3d_fwd(inputs_shape, filters_shape, subsample, def run_conv3d_fwd(inputs_shape, filters_shape, subsample,
border_mode, conv_mode): border_mode, conv_mode):
...@@ -1064,6 +1077,7 @@ def test_conv3d_bwd(): ...@@ -1064,6 +1077,7 @@ def test_conv3d_bwd():
if not dnn.dnn_available(test_ctx_name): if not dnn.dnn_available(test_ctx_name):
raise SkipTest(dnn.dnn_available.msg) raise SkipTest(dnn.dnn_available.msg)
utt.seed_rng()
def run_conv3d_bwd(inputs_shape, filters_shape, subsample, def run_conv3d_bwd(inputs_shape, filters_shape, subsample,
border_mode, conv_mode): border_mode, conv_mode):
...@@ -1783,6 +1797,10 @@ def test_dnn_batchnorm_valid_and_invalid_axes(): ...@@ -1783,6 +1797,10 @@ def test_dnn_batchnorm_valid_and_invalid_axes():
def test_dnn_rnn_gru(): def test_dnn_rnn_gru():
if not dnn.dnn_available(test_ctx_name):
raise SkipTest(dnn.dnn_available.msg)
utt.seed_rng()
# test params # test params
input_dim = 32 input_dim = 32
hidden_dim = 16 hidden_dim = 16
...@@ -1882,6 +1900,10 @@ def test_dnn_rnn_gru(): ...@@ -1882,6 +1900,10 @@ def test_dnn_rnn_gru():
def test_dnn_rnn_gru_bidi(): def test_dnn_rnn_gru_bidi():
if not dnn.dnn_available(test_ctx_name):
raise SkipTest(dnn.dnn_available.msg)
utt.seed_rng()
# test params # test params
input_dim = 32 input_dim = 32
hidden_dim = 16 hidden_dim = 16
...@@ -1931,6 +1953,10 @@ def test_dnn_rnn_gru_bidi(): ...@@ -1931,6 +1953,10 @@ def test_dnn_rnn_gru_bidi():
def test_dnn_rnn_lstm(): def test_dnn_rnn_lstm():
if not dnn.dnn_available(test_ctx_name):
raise SkipTest(dnn.dnn_available.msg)
utt.seed_rng()
# test params # test params
input_dim = 32 input_dim = 32
hidden_dim = 16 hidden_dim = 16
...@@ -2006,6 +2032,10 @@ def test_dnn_rnn_lstm(): ...@@ -2006,6 +2032,10 @@ def test_dnn_rnn_lstm():
def test_dnn_rnn_lstm_grad_c(): def test_dnn_rnn_lstm_grad_c():
if not dnn.dnn_available(test_ctx_name):
raise SkipTest(dnn.dnn_available.msg)
utt.seed_rng()
# test params # test params
input_dim = 32 input_dim = 32
hidden_dim = 16 hidden_dim = 16
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论