diff --git a/demo/sequence-labeling/run_sequence_label.sh b/demo/sequence-labeling/run_sequence_label.sh index 1b70a0de116cc940db26b670be04c6fdcc486773..55ca0bd3361ba1fed725ef00feb0f9efaef6b5ca 100644 --- a/demo/sequence-labeling/run_sequence_label.sh +++ b/demo/sequence-labeling/run_sequence_label.sh @@ -1,4 +1,4 @@ -export CUDA_VISIBLE_DEVICES=1 +export CUDA_VISIBLE_DEVICES=0 CKPT_DIR="./ckpt_sequence_label" python -u sequence_label.py \ diff --git a/demo/text-classification/run_classifier.sh b/demo/text-classification/run_classifier.sh index d8cfe11e81a6b4aed96798447bc42868e24fe84e..0b53a268c31929fd810138bf0fa36ab41469fc60 100644 --- a/demo/text-classification/run_classifier.sh +++ b/demo/text-classification/run_classifier.sh @@ -1,4 +1,4 @@ -export CUDA_VISIBLE_DEVICES=5 +export CUDA_VISIBLE_DEVICES=0 # User can select chnsenticorp, nlpcc_dbqa, lcqmc for different task DATASET="chnsenticorp" diff --git a/demo/text-classification/run_predict.sh b/demo/text-classification/run_predict.sh index bc664cb8bf315452cf606b63fe227f8971726671..1fe8680f44fea98a743d74507799abaea91757ca 100644 --- a/demo/text-classification/run_predict.sh +++ b/demo/text-classification/run_predict.sh @@ -1,4 +1,4 @@ -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