diff --git a/demo/text-classification/run_classifier.sh b/demo/text-classification/run_classifier.sh index d1aec212f97f2bbcc22b8f24a16bdce321099e49..408d6fe8c1b7fda34a0c5202b580cd8aff4a0444 100644 --- a/demo/text-classification/run_classifier.sh +++ b/demo/text-classification/run_classifier.sh @@ -1,5 +1,5 @@ export FLAGS_eager_delete_tensor_gb=0.0 -export CUDA_VISIBLE_DEVICES=0 +export CUDA_VISIBLE_DEVICES=1,2,3,4 # User can select chnsenticorp, nlpcc_dbqa, lcqmc and so on for different task DATASET="chnsenticorp"