提交 3851f020 authored 作者: Nan Rosemary Ke's avatar Nan Rosemary Ke

bug fix

上级 bc186ecc
...@@ -1379,14 +1379,14 @@ class ProfileStats(object): ...@@ -1379,14 +1379,14 @@ class ProfileStats(object):
" for inputs (%s). Make sure that the inputs have same " " for inputs (%s). Make sure that the inputs have same "
" dtype." % [i.type for i in node.inputs], file=file) " dtype." % [i.type for i in node.inputs], file=file)
printed_tip = True printed_tip = True
# tip 7 # tip 7
import theano.sandbox.cuda as cuda import theano.sandbox.cuda as cuda
from theano.tensor.nnet import LogSoftmax from theano.tensor.nnet import LogSoftmax
import theano.tensor.signal.pool as pool import theano.tensor.signal.pool as pool
import theano.gpuarray import theano.gpuarray
for a, t in iteritems(apply_time): for a, t in iteritems(self.apply_time):
node = a[1] node = a[1]
if (isinstance(node.op, pool.Pool)): if (isinstance(node.op, pool.Pool)):
if (not cuda.dnn.dnn_available() and not theano.gpuarray.dnn.dnn_present()): if (not cuda.dnn.dnn_available() and not theano.gpuarray.dnn.dnn_present()):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论