Add gflags

7 years ago · 5162c41a92
parent 57b973ae76
commit 5162c41a92
1 changed files with 1 additions and 1 deletions
--- 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]] +