提交 60b6facf authored 作者: Frederic Bastien's avatar Frederic Bastien

don't hide bug in profile mode.

上级 1e8c3a25
......@@ -16,11 +16,8 @@ logging.getLogger('theano.gradient').setLevel(logging.INFO)
def get_mode():
return None if theano.compile.default_mode != "PROFILE_MODE" else theano.compile.ProfileMode()
def print_mode(mode):
try:
if mode != None:
if mode != None and isinstance(mode,(theano.compile.ProfileMode,)):
mode.print_summary()
except:
pass
def run_nnet(use_gpu):
n_batch = 16
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论