run_qa.py add xpu chioce of device,*test=kunlun (#3046)

* run_qa.py add xpu chioce of device,*test=kunlun * run_qa.py add xpu chioce of device,*test=kunlun
PaddlePaddle · Aug 16, 2022 · 56a8957 · 56a8957
1 parent 42ec8e3
commit 56a8957
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/model_zoo/ernie-3.0/run_qa.py b/model_zoo/ernie-3.0/run_qa.py
@@ -132,7 +132,7 @@ def parse_args():
                         help="random seed for initialization")
     parser.add_argument(
         '--device',
-        choices=['cpu', 'gpu'],
+        choices=['cpu', 'gpu', 'xpu'],
         default="gpu",
         help="Select which device to train model, defaults to gpu.")
     parser.add_argument(