提交 f1ad528f authored 作者: Roy Xue's avatar Roy Xue

update test file with new config setting

上级 7717caa8
...@@ -12,11 +12,13 @@ import theano.tensor as T ...@@ -12,11 +12,13 @@ import theano.tensor as T
def test_profiling(): def test_profiling():
old1 = theano.config.profile config1 = theano.config.profile
old2 = theano.config.profile_memory config2 = theano.config.profile_memory
config3 = theano.config.profiling.min_peak_memory
try: try:
theano.config.profile = True theano.config.profile = True
theano.config.profile_memory = True theano.config.profile_memory = True
theano.config.profiling.min_peak_memory = True
x = [T.dvector("val%i" % i) for i in range(3)] x = [T.dvector("val%i" % i) for i in range(3)]
...@@ -46,8 +48,9 @@ def test_profiling(): ...@@ -46,8 +48,9 @@ def test_profiling():
assert "Minimum peak from all valid apply node order is 8192KB" in the_string assert "Minimum peak from all valid apply node order is 8192KB" in the_string
finally: finally:
theano.config.profile = old1 theano.config.profile = config1
theano.config.profile_memory = old2 theano.config.profile_memory = config2
theano.config.profiling.min_peak_memory = config3
if __name__ == '__main__': if __name__ == '__main__':
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论