提交 63a7a5dc authored 作者: Pierre Luc Carrier's avatar Pierre Luc Carrier

Make tests in TestPushOutSumOfDot faster to avoid timeout in Travis with FAST_COMPILE

上级 aaa13b11
...@@ -284,13 +284,15 @@ class TestPushOutSumOfDot(): ...@@ -284,13 +284,15 @@ class TestPushOutSumOfDot():
This test case comes from https://github.com/rizar/scan-grad-speed and This test case comes from https://github.com/rizar/scan-grad-speed and
is an example of actual computation done with scan in the context of is an example of actual computation done with scan in the context of
machine translation machine translation
'dim' has been reduced from 1000 to 5 to make the test run faster
""" """
# Parameters from an actual machine tranlation run # Parameters from an actual machine tranlation run
batch_size = 80 batch_size = 80
seq_len = 50 seq_len = 50
n_words = 80 * 50 n_words = 80 * 50
dim = 1000 dim = 5
# Weight matrices # Weight matrices
U = theano.shared(numpy.random.normal(size=(dim, dim), U = theano.shared(numpy.random.normal(size=(dim, dim),
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论