Update run_classifier.sh

57e49059 · Steffy-zxf · GitHub · cb9d9076 · 57e49059
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

demo/text-classification/run_classifier.sh demo/text-classification/run_classifier.sh +1 -1

未找到文件。
--- a/demo/text-classification/run_classifier.sh
+++ b/demo/text-classification/run_classifier.sh
 export FLAGS_eager_delete_tensor_gb=0.0
-export CUDA_VISIBLE_DEVICES=1,2,3,4
+export CUDA_VISIBLE_DEVICES=0
 # User can select chnsenticorp, nlpcc_dbqa, lcqmc and so on for different task
 DATASET="chnsenticorp"