提交 dbc61422 authored 作者: Frederic's avatar Frederic

Fix test in DebugMode

上级 39673cc9
......@@ -35,5 +35,9 @@ def test_none_Constant():
x = tensor.vector('x')
y = tensor.argmax(x)
f = theano.function([x], [y])
kwargs = {}
# We can't pickle DebugMode
if theano.config.mode in ["DebugMode", "DEBUG_MODE"]:
kwargs = {'mode': 'FAST_RUN'}
f = theano.function([x], [y], **kwargs)
cPickle.loads(cPickle.dumps(f))
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论