提交 815f3345 authored 作者: Frederic's avatar Frederic

Fix profiling print to the right output

上级 681c67fa
......@@ -697,7 +697,7 @@ class ProfileStats(object):
print('Time in all call to theano.grad() %es' %
theano.gradient.grad_time, file=file)
total_time = time.time() - theano_imported_time
print('Time since theano import %.3fs' % (total_time))
print('Time since theano import %.3fs' % (total_time), file=file)
def summary_memory(self, file, N=None):
fct_memory = {} # fgraph->dict(node->[outputs size])
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论