Add gflags

5162c41a · Yang Yu · 57b973ae · 5162c41a
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

python/paddle/v2/fluid/__init__.py python/paddle/v2/fluid/__init__.py +1 -1

未找到文件。
--- a/python/paddle/v2/fluid/__init__.py
+++ b/python/paddle/v2/fluid/__init__.py
@@ -36,7 +36,7 @@ def __read_gflags_from_env__():
    """
    import sys
    import core
-    read_env_flags = ['use_pinned_memory']
+    read_env_flags = ['use_pinned_memory', 'check_nan_inf']
    if core.is_compile_gpu():
        read_env_flags.append('fraction_of_gpu_memory_to_use')
    core.init_gflags([sys.argv[0]] +