提交 89082669 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Fix mode settings.

上级 6811ef04
......@@ -103,7 +103,10 @@ def test_conv3d(border_mode):
if ndimage is None or not theano.config.cxx:
raise SkipTest("conv3d2d tests need SciPy and a c++ compiler")
mode = theano.compile.mode.get_default_mode('FAST_RUN')
if theano.config.mode == 'FAST_COMPILE':
mode = theano.compile.mode.get_mode('FAST_RUN')
else:
mode = theano.compile.mode.get_default_mode()
shared = theano.tensor._shared
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论