From 0d987a6ff3680fe9c47c3c598387daecc2da1699 Mon Sep 17 00:00:00 2001 From: wuzewu Date: Tue, 16 Apr 2019 10:29:47 +0800 Subject: [PATCH] update demo script --- demo/sequence-labeling/run_sequence_label.sh | 2 +- demo/text-classification/run_classifier.sh | 2 +- demo/text-classification/run_predict.sh | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/demo/sequence-labeling/run_sequence_label.sh b/demo/sequence-labeling/run_sequence_label.sh index 1b70a0de..55ca0bd3 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 d8cfe11e..0b53a268 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 bc664cb8..1fe8680f 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 -- GitLab