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

move tips to profiling.py

上级 c4e4acca
...@@ -694,21 +694,21 @@ Test them first, as they are not guaranteed to always provide a speedup.""") ...@@ -694,21 +694,21 @@ Test them first, as they are not guaranteed to always provide a speedup.""")
break break
# tip 6 # tip 6
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(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")
if not printed_tip: if not printed_tip:
print(" Sorry, no tip for today.") print(" Sorry, no tip for today.")
......
...@@ -1380,23 +1380,23 @@ class ProfileStats(object): ...@@ -1380,23 +1380,23 @@ class ProfileStats(object):
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(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")
# printed_tip = True printed_tip = True
# 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")
# printed_tip = True printed_tip = True
if not printed_tip: if not printed_tip:
print(" Sorry, no tip for today.", file=file) print(" Sorry, no tip for today.", file=file)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论