提交 b99ee905 authored 作者: Frédéric Bastien's avatar Frédéric Bastien

Correct mode fix

上级 815e2785
......@@ -835,11 +835,11 @@ def conv_grad(mode, bs, ch, nf, rImg1, rImg2, rFlt1, rFlt2, subsample, op):
outputs.extend([corr_op_dik, conv_op_dik,
corr_op_dki, conv_op_dki])
theano_mode = None
if theano.config.mode == 'DEBUG_MODE':
# TODO: fix when the abstractconv tests can pass debug mode.
theano_mode = 'FAST_RUN'
f = theano.function([i, k], outputs, mode=theano_mode)
mode = theano_mode
if theano.config.mode == 'DEBUG_MODE':
mode = theano.compile.mode.get_mode('FAST_RUN').including('gpu')
f = theano.function([i, k], outputs, mode=mode)
allvals = f(npy_img, npy_kern)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论