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

flake 8 compatible

上级 713ed61f
...@@ -703,11 +703,11 @@ Test them first, as they are not guaranteed to always provide a speedup.""") ...@@ -703,11 +703,11 @@ Test them first, as they are not guaranteed to always provide a speedup.""")
if (isinstance(node.op, pool.Pool)): if (isinstance(node.op, pool.Pool)):
if not cuda.dnn.dnn_available(): if not cuda.dnn.dnn_available():
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(): if not cuda.dnn.dnn_available():
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")
if not printed_tip: if not printed_tip:
print(" Sorry, no tip for today.") print(" Sorry, no tip for today.")
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论