提交 e8c47f1e authored 作者: Christof Angermueller's avatar Christof Angermueller

Suppress profiling output

上级 51c6cb78
......@@ -33,7 +33,8 @@ class TestD3Viz(unittest.TestCase):
def test_mlp_profiled(self):
m = models.Mlp()
f = th.function(m.inputs, m.outputs, profile=True)
profile = th.compile.profiling.ProfileStats(False)
f = th.function(m.inputs, m.outputs, profile=profile)
x_val = self.rng.normal(0, 1, (1000, m.nfeatures))
f(x_val)
self.check(f)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论