Fix python.v2.fluid arg parse (#6055)

* fix python gflags init * format code
PaddlePaddle · Nov 30, 2017 · 82dd165 · 82dd165
1 parent dc91c4e
commit 82dd165
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/python/paddle/v2/fluid/__init__.py b/python/paddle/v2/fluid/__init__.py
@@ -36,7 +36,8 @@ def __read_gflags_from_env__():
     read_env_flags = ['use_pinned_memory']
     if core.is_compile_gpu():
         read_env_flags.append('fraction_of_gpu_memory_to_use')
-    core.init_gflags(sys.argv + ["--tryfromenv=" + ",".join(read_env_flags)])
+    core.init_gflags([sys.argv[0]] +
+                     ["--tryfromenv=" + ",".join(read_env_flags)])
 
 
 __read_gflags_from_env__()