提交 331304fe authored 作者: Nan Rosemary Ke's avatar Nan Rosemary Ke

flake8 bug fix

上级 f740c9b8
...@@ -701,11 +701,11 @@ Test them first, as they are not guaranteed to always provide a speedup.""") ...@@ -701,11 +701,11 @@ Test them first, as they are not guaranteed to always provide a speedup.""")
for a, t in iteritems(apply_time): for a, t in iteritems(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()):
print("Install CuDNN to do pooling faster" print("Install CuDNN to do pooling faster"
"this allows the operation to run on GPU") "this allows the operation to run on GPU")
if (isinstance(node.op, LogSoftmax)): if (isinstance(node.op, LogSoftmax)):
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()):
print("Install CuDNN to do LogSoftmax faster" print("Install CuDNN to do LogSoftmax faster"
"this allows the operation to run on GPU") "this allows the operation to run on GPU")
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论