refine the device choose test=develop (#4247)

f9853b7f · wangchaochaohu · GitHub · cb0dcc12 · f9853b7f
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

PaddleNLP/sentiment_classification/run_ernie.sh PaddleNLP/sentiment_classification/run_ernie.sh +1 -1

未找到文件。
--- a/PaddleNLP/sentiment_classification/run_ernie.sh
+++ b/PaddleNLP/sentiment_classification/run_ernie.sh
@@ -2,7 +2,7 @@
 export FLAGS_fraction_of_gpu_memory_to_use=0.95
 export FLAGS_enable_parallel_graph=1
 export FLAGS_sync_nccl_allreduce=1
-export CUDA_VISIBLE_DEVICES=12
+export CUDA_VISIBLE_DEVICES=0
 export CPU_NUM=1
 ERNIE_PRETRAIN=./ernie_pretrain_model/
 DATA_PATH=./senta_data