提交 5fc2af74 authored 作者: Frederic Bastien's avatar Frederic Bastien

make ProfileMode use the same default optimizer as the one in default_mode!

上级 50ca43a3
...@@ -37,6 +37,7 @@ predefined_linkers = { ...@@ -37,6 +37,7 @@ predefined_linkers = {
'c&py' : gof.DualLinker(checker = check_equal) 'c&py' : gof.DualLinker(checker = check_equal)
} }
#Keep default_linker the same as the one for default_mode
default_linker = 'c|py' default_linker = 'c|py'
def register_linker(name, linker): def register_linker(name, linker):
...@@ -63,7 +64,8 @@ predefined_optimizers = { ...@@ -63,7 +64,8 @@ predefined_optimizers = {
'fast_run_stable' : OPT_FAST_RUN_STABLE, 'fast_run_stable' : OPT_FAST_RUN_STABLE,
'fast_compile' : OPT_FAST_COMPILE 'fast_compile' : OPT_FAST_COMPILE
} }
default_optimizer = 'merge' #Keep default_optimizer the same as the one for default_mode
default_optimizer = 'fast_run'
def register_optimizer(name, opt): def register_optimizer(name, opt):
"""Add a `Optimizer` which can be referred to by `name` in `Mode`.""" """Add a `Optimizer` which can be referred to by `name` in `Mode`."""
...@@ -157,6 +159,7 @@ predefined_modes = {'FAST_COMPILE': FAST_COMPILE, ...@@ -157,6 +159,7 @@ predefined_modes = {'FAST_COMPILE': FAST_COMPILE,
# The default mode used by functions and modules is read from the environment # The default mode used by functions and modules is read from the environment
# variable THEANO_DEFAULT_MODE. Unit tests will run using this value. If the env. var. # variable THEANO_DEFAULT_MODE. Unit tests will run using this value. If the env. var.
# is not set, it will default to 'FAST_RUN' # is not set, it will default to 'FAST_RUN'
# keep default_mode.optimizer==default_optimizer and default_mode.linker==default_linker!
## ##
default_mode = os.getenv('THEANO_DEFAULT_MODE','FAST_RUN') default_mode = os.getenv('THEANO_DEFAULT_MODE','FAST_RUN')
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论