diff --git a/fluid/PaddleCV/object_detection/main_quant.py b/fluid/PaddleCV/object_detection/main_quant.py index 0fde0a78b7c99fb2fe6c5facff75c3300ddc00ab..bd7d377e69e95dcaf066a40941cf48091583e7ab 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,