run.sh 663 字节
Newer Older
X
xuezhong 已提交
1
export CUDA_VISIBLE_DEVICES=0
X
xuezhong 已提交
2
python run.py   \
Q
qiuxuezhong 已提交
3 4 5 6 7 8 9 10 11 12 13
--trainset 'data/preprocessed/trainset/search.train.json' \
           'data/preprocessed/trainset/zhidao.train.json' \
--devset 'data/preprocessed/devset/search.dev.json' \
         'data/preprocessed/devset/zhidao.dev.json' \
--testset 'data/preprocessed/testset/search.test.json' \
          'data/preprocessed/testset/zhidao.test.json' \
--vocab_dir 'data/vocab' \
--use_gpu true \
--save_dir ./models \
--pass_num 10 \
--learning_rate 0.001 \
X
xuezhong 已提交
14
--batch_size 32 \
Q
qiuxuezhong 已提交
15 16 17 18 19 20
--embed_size 300 \
--hidden_size 150 \
--max_p_num 5 \
--max_p_len 500 \
--max_q_len 60 \
--max_a_len 200 \
X
xuezhong 已提交
21
--weight_decay 0.0 \
X
xuezhong 已提交
22
--drop_rate 0.2 $@\