提交 e8b30662 authored 作者: Frederic's avatar Frederic

Don't print test profile at process exit.

上级 6922c2f9
...@@ -18,7 +18,8 @@ def test_profiling(): ...@@ -18,7 +18,8 @@ def test_profiling():
x = T.dvector("x") x = T.dvector("x")
y = T.dvector("y") y = T.dvector("y")
z = x + y z = x + y
f = theano.function([x, y], z, profile=True, name="test_profiling") p = theano.ProfileStats(False)
f = theano.function([x, y], z, profile=p, name="test_profiling")
output = f([1, 2, 3, 4], [1, 1, 1, 1]) output = f([1, 2, 3, 4], [1, 1, 1, 1])
buf = StringIO.StringIO() buf = StringIO.StringIO()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论