提交 1a579783 authored 作者: Frederic Bastien's avatar Frederic Bastien

added a fct to return the instance of the default mode. It check that it exist…

added a fct to return the instance of the default mode. It check that it exist and raise an informative error if it don't exist.
上级 2f397de7
...@@ -213,6 +213,11 @@ predefined_modes = {'FAST_COMPILE': FAST_COMPILE, ...@@ -213,6 +213,11 @@ predefined_modes = {'FAST_COMPILE': FAST_COMPILE,
## ##
default_mode = os.getenv('THEANO_DEFAULT_MODE','FAST_RUN') default_mode = os.getenv('THEANO_DEFAULT_MODE','FAST_RUN')
def get_default_mode():
if not predefined_modes.has_key(default_mode):
raise Exception("No predefixed mode exist for string: %s"%default_mode)
return predefined_modes[default_mode]
def register_mode(name, mode): def register_mode(name, mode):
"""Add a `Mode` which can be referred to by `name` in `function`.""" """Add a `Mode` which can be referred to by `name` in `function`."""
if name in predefined_modes: if name in predefined_modes:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论