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

Fix mode settings.

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