提交 c024a85e authored 作者: James Bergstra's avatar James Bergstra

environment var configuration of DEBUG_MODE

上级 c11b64ad
"""Provides `DebugMode`, an evaluation mode for debugging theano internals.""" """Provides `DebugMode`, an evaluation mode for debugging theano internals."""
__docformat__ = "restructuredtext en" __docformat__ = "restructuredtext en"
import time, copy, sys, copy_reg, gc import time, copy, sys, copy_reg, gc, os
from StringIO import StringIO from StringIO import StringIO
import numpy import numpy
...@@ -1352,27 +1352,27 @@ class DebugMode(Mode): ...@@ -1352,27 +1352,27 @@ class DebugMode(Mode):
""" """
stability_patience = 10 stability_patience = int(os.getenv('THEANO_DEBUGMODE_PATIENCE', 10))
""" """
When checking for the stability of optimization, recompile the graph this many times. When checking for the stability of optimization, recompile the graph this many times.
""" """
check_c_code = True check_c_code = bool(int(os.getenv('THEANO_DEBUGMODE_CHECK_C', 1)))
""" """
Should we evaluate (and check) the `c_code` implementations? Should we evaluate (and check) the `c_code` implementations?
""" """
check_py_code = True check_py_code = bool(int(os.getenv('THEANO_DEBUGMODE_CHECK_PY', 1)))
""" """
Should we evaluate (and check) the `perform` implementations? Should we evaluate (and check) the `perform` implementations?
""" """
check_isfinite = True check_isfinite = bool(int(os.getenv('THEANO_DEBUGMODE_CHECK_FINITE', 1)))
""" """
Should we check for (and complain about) NaN/Inf ndarray elements? Should we check for (and complain about) NaN/Inf ndarray elements?
""" """
require_matching_strides = False require_matching_strides = bool(int(os.getenv('THEANO_DEBUGMODE_CHECK_STRIDES', 0)))
""" """
Should we check for (and complain about) Ops whose python and C outputs are ndarrays with Should we check for (and complain about) Ops whose python and C outputs are ndarrays with
different strides? (This can catch bugs, but is generally overly strict.) different strides? (This can catch bugs, but is generally overly strict.)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论