提交 896b68ee authored 作者: Nan Rosemary Ke's avatar Nan Rosemary Ke

bug fix

上级 b0f2c6df
......@@ -1386,8 +1386,8 @@ class ProfileStats(object):
import theano.tensor.signal.pool as pool
import theano.gpuarray
for a, t in iteritems(self.apply_time):
node = a[1]
for a in self.apply_time):
node = a
if (isinstance(node.op, pool.Pool)):
if (not cuda.dnn.dnn_available() and not theano.gpuarray.dnn.dnn_present()):
print("Install CuDNN to do pooling faster"
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论