提交 386e5aa0 authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Use mode=FAST_RUN in scan as well

This should prevent an 10 minutes timeout in FAST_COMPILE
上级 69bcff08
...@@ -292,7 +292,8 @@ class test_RopLop(RopLop_checker): ...@@ -292,7 +292,8 @@ class test_RopLop(RopLop_checker):
sy, _ = theano.scan(lambda i, y, x, v: sy, _ = theano.scan(lambda i, y, x, v:
(tensor.grad(y[i], x) * v).sum(), (tensor.grad(y[i], x) * v).sum(),
sequences=tensor.arange(a_pooled.shape[0]), sequences=tensor.arange(a_pooled.shape[0]),
non_sequences=[a_pooled, x, ex]) non_sequences=[a_pooled, x, ex],
mode=mode)
scan_f = function([], sy, on_unused_input='ignore', mode=mode) scan_f = function([], sy, on_unused_input='ignore', mode=mode)
v1 = rop_f() v1 = rop_f()
v2 = scan_f() v2 = scan_f()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论