forked from donglixp/coarse2fine
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathrun_train.sh
executable file
·45 lines (39 loc) · 1.11 KB
/
run_train.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
#!/bin/bash
dataset=$1
base_dir=$(dirname "$(readlink -f $0)")
data_dir=${base_dir}/data_model/${dataset}
emb_dir=${base_dir}/../embeddings
emb_exp_dir=${emb_dir}/2019-05-14_18-19-19-python-so-200
python src/train.py \
-root_dir ${base_dir}/data_model/ \
-dataset ${dataset} \
-rnn_size 300 \
-decoder_input_size 200 \
-layers 1 \
-learning_rate 0.002 \
-epochs 100 \
-batch_size 12 \
-global_attention "dot" \
-attn_hidden 0 \
-dropout 0.3 \
-dropout_i 0.3 \
-lock_dropout \
-copy_prb hidden \
-batch_report_every 25 \
-start_checkpoint_at 10 \
-use_custom_embeddings \
-vocab_file ${emb_exp_dir}/*.vocab \
-pt_embeddings ${emb_dir}/glove.6B.200d.txt \
-word_embeddings ${emb_exp_dir}/*.emb \
-word_emb_size 200 \
-pt_factor 0.1 \
-ft_factor 0.9 \
-cuda \
-exp_name django-orig-pyso-200-embeddings \
# -seed 1234 \
# -use_custom_embeddings \
# -vocab_file ${emb_exp_dir}/*.vocab \
# -pt_embeddings ${emb_dir}/glove.6B.50d.txt \
# -word_embeddings ${emb_exp_dir}/*.emb \
# -pt_factor 0.1 \
# -ft_factor 0.9 \