update demo script

0d987a6f · wuzewu · 6beb0ed2 · 0d987a6f · 0d987a6f · 0d987a6f
3 changed file
--- a/demo/sequence-labeling/run_sequence_label.sh
+++ b/demo/sequence-labeling/run_sequence_label.sh
-export CUDA_VISIBLE_DEVICES=1
+export CUDA_VISIBLE_DEVICES=0

 CKPT_DIR="./ckpt_sequence_label"
 python -u sequence_label.py \

--- a/demo/text-classification/run_classifier.sh
+++ b/demo/text-classification/run_classifier.sh
-export CUDA_VISIBLE_DEVICES=5
+export CUDA_VISIBLE_DEVICES=0

 # User can select chnsenticorp, nlpcc_dbqa, lcqmc for different task
 DATASET="chnsenticorp"

--- a/demo/text-classification/run_predict.sh
+++ b/demo/text-classification/run_predict.sh
-export CUDA_VISIBLE_DEVICES=5
+export CUDA_VISIBLE_DEVICES=0

 CKPT_DIR="./ckpt_20190414203357/best_model"
 python -u predict.py --checkpoint_dir $CKPT_DIR --max_seq_len 128