From 7852d0604d9c4d82778ef1c705dbcfd904eecadf Mon Sep 17 00:00:00 2001
From: qingqing01 <dangqingqing@baidu.com>
Date: Fri, 16 Nov 2018 13:35:45 +0800
Subject: [PATCH] Fix bug in some Python env. (#1457)

* Fix bug in some Python env.
---
 fluid/PaddleCV/object_detection/main_quant.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/fluid/PaddleCV/object_detection/main_quant.py b/fluid/PaddleCV/object_detection/main_quant.py
index 0fde0a78..bd7d377e 100644
--- a/fluid/PaddleCV/object_detection/main_quant.py
+++ b/fluid/PaddleCV/object_detection/main_quant.py
@@ -5,6 +5,7 @@ import argparse
 import functools
 import shutil
 import math
+import multiprocessing
 
 import paddle
 import paddle.fluid as fluid
@@ -127,7 +128,7 @@ def train(args,
 
     if parallel:
         train_exe = fluid.ParallelExecutor(main_program=train_prog,
-            use_cuda=use_gpu, loss_name=loss.name)
+            use_cuda=True if use_gpu else False, loss_name=loss.name)
 
     train_reader = reader.train(data_args,
                                 train_file_list,
-- 
GitLab