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

ccw4102

上级 167c7d86
......@@ -13,9 +13,6 @@ from theano.compile.mode import (Mode, register_mode,
predefined_optimizers)
from theano.configparser import config
from theano.compile.function_module import FunctionMaker
import theano.sandbox.cuda as cuda
from theano.tensor.nnet import LogSoftmax
import theano.tensor.signal.pool as pool
from .profiling import ProfileStats
......@@ -695,7 +692,12 @@ Test them first, as they are not guaranteed to always provide a speedup.""")
print(" - MRG_RandomStreams is the only random number"
" generator supported on the GPU.")
break
# tip 6
import theano.sandbox.cuda as cuda
from theano.tensor.nnet import LogSoftmax
import theano.tensor.signal.pool as pool
for a, t in iteritems(apply_time):
node = a[1]
if (isinstance(node.op, pool.Pool)):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论