提交 984404ff authored 作者: Brandon T. Willard's avatar Brandon T. Willard

Remove reference to theano.compile.mode.predefined_nodes

上级 9f7386ce
...@@ -596,16 +596,15 @@ AddConfigVar( ...@@ -596,16 +596,15 @@ AddConfigVar(
# Also, please be careful not to modify the first item in the enum when adding # Also, please be careful not to modify the first item in the enum when adding
# new modes, since it is the default mode. # new modes, since it is the default mode.
def filter_mode(val): def filter_mode(val):
if ( if val in [
val "Mode",
in [ "DebugMode",
"Mode", "FAST_RUN",
"DebugMode", "NanGuardMode",
"NanGuardMode", "FAST_COMPILE",
"DEBUG_MODE", "DEBUG_MODE",
] "JAX",
or val in theano.compile.mode.predefined_modes ]:
):
return val return val
# This can be executed before Theano is completly imported, so # This can be executed before Theano is completly imported, so
# theano.Mode is not always available. # theano.Mode is not always available.
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论