提交 0504c377 authored 作者: Frederic's avatar Frederic 提交者: AlOa

Don't override THEANO_FLAGS

上级 dee4c417
......@@ -29,9 +29,10 @@ if __name__ == '__main__':
if hasattr(options, "help"):
print options.help
sys.exit(0)
os.environ['THEANO_FLAGS'] = 'openmp=false'
orig_flags = os.environ.get('THEANO_FLAGS', '')
os.environ['THEANO_FLAGS'] = orig_flags + ',openmp=false'
(cheapTime, costlyTime) = runScript(N=options.N)
os.environ['THEANO_FLAGS'] = 'openmp=true'
os.environ['THEANO_FLAGS'] = orig_flags + ',openmp=true'
(cheapTimeOpenmp, costlyTimeOpenmp) = runScript(N=options.N)
if cheapTime > cheapTimeOpenmp:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论