提交 0d987a6f 编写于 作者: W wuzewu

update demo script

上级 6beb0ed2
export CUDA_VISIBLE_DEVICES=1 export CUDA_VISIBLE_DEVICES=0
CKPT_DIR="./ckpt_sequence_label" CKPT_DIR="./ckpt_sequence_label"
python -u sequence_label.py \ python -u sequence_label.py \
......
export CUDA_VISIBLE_DEVICES=5 export CUDA_VISIBLE_DEVICES=0
# User can select chnsenticorp, nlpcc_dbqa, lcqmc for different task # User can select chnsenticorp, nlpcc_dbqa, lcqmc for different task
DATASET="chnsenticorp" DATASET="chnsenticorp"
......
export CUDA_VISIBLE_DEVICES=5 export CUDA_VISIBLE_DEVICES=0
CKPT_DIR="./ckpt_20190414203357/best_model" CKPT_DIR="./ckpt_20190414203357/best_model"
python -u predict.py --checkpoint_dir $CKPT_DIR --max_seq_len 128 python -u predict.py --checkpoint_dir $CKPT_DIR --max_seq_len 128
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册