提交 a23392bf authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Fix test when run with DebugMode.patience=1.

上级 ee7592e7
...@@ -264,7 +264,10 @@ def test_stochasticoptimization(): ...@@ -264,7 +264,10 @@ def test_stochasticoptimization():
try: try:
theano.function([a, b], theano.function([a, b],
theano.tensor.add(a, b), theano.tensor.add(a, b),
mode=debugmode.DebugMode(optimizer=opt, check_c_code=True)) mode=debugmode.DebugMode(
optimizer=opt,
check_c_code=True,
stability_patience=max(2, config.DebugMode.patience)))
except debugmode.StochasticOrder: except debugmode.StochasticOrder:
return # TEST PASS return # TEST PASS
assert False assert False
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论