提交 9ef52f35 authored 作者: sebastien-j's avatar sebastien-j

Speed up scan test

上级 4d8536b5
...@@ -1911,9 +1911,11 @@ class T_Scan(unittest.TestCase): ...@@ -1911,9 +1911,11 @@ class T_Scan(unittest.TestCase):
for param, gparam in zip(params, gparams)] for param, gparam in zip(params, gparams)]
learn_rnn_fn = theano.function(inputs=[x, t], learn_rnn_fn = theano.function(inputs=[x, t],
outputs=cost, outputs=cost,
updates=updates) updates=updates,
mode="FAST_RUN")
eval_rnn_fn = theano.function(inputs=[x], eval_rnn_fn = theano.function(inputs=[x],
outputs=y) outputs=y,
mode="FAST_RUN")
# artificial data # artificial data
x_v = numpy.arange(0., 100., 0.21, dtype=theano.config.floatX) x_v = numpy.arange(0., 100., 0.21, dtype=theano.config.floatX)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论