diff --git a/PaddleNLP/dialogue_domain_classification/run_classifier.py b/PaddleNLP/dialogue_domain_classification/run_classifier.py index b04574dfb7eb9af60115bef995c1d37e62c447c4..56f8dc90b7bbe33b194f5db85667de542f81036f 100755 --- a/PaddleNLP/dialogue_domain_classification/run_classifier.py +++ b/PaddleNLP/dialogue_domain_classification/run_classifier.py @@ -399,8 +399,8 @@ def main(args): place = fluid.cuda_places() DEV_COUNT = fluid.core.get_cuda_device_count() else: - place = fluid.cpu_places() os.environ['CPU_NUM'] = str(args.cpu_num) + place = fluid.cpu_places() DEV_COUNT = args.cpu_num logger.info("Dev Num is %s" % str(DEV_COUNT)) exe = fluid.Executor(place[0])