-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathcommands.sh
22 lines (21 loc) · 1.18 KB
/
commands.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
python preprocess.py \
--train_src 'data/processed_data/eng/train_src' \
--train_tgt 'data/processed_data/eng/train_tgt' \
--eval_src 'data/processed_data/eng/eval_src' \
--eval_tgt 'data/processed_data/eng/dev.reference0' \
--test_src 'data/processed_data/eng/test_src' \
--spl_sym 'data/processed_data/special_symbols' \
--model gat --lang eng --sentencepiece True \
--vocab_size 16000 --sentencepiece_model 'bpe'
python train_single.py \
--train_path 'data/processed_graphs/eng/gat/train' \
--eval_path 'data/processed_graphs/eng/gat/eval' \
--test_path 'data/processed_graphs/eng/gat/test' \
--src_vocab 'vocabs/gat/eng/src_vocab' \
--tgt_vocab 'vocabs/gat/eng/train_vocab.model' \
--batch_size 1 --enc_type gat --dec_type transformer --model gat --vocab_size 16000 \
--emb_dim 16 --hidden_size 16 --filter_size 16 --beam_size 5 \
--beam_alpha 0.1 --enc_layers 1 --dec_layers 1 --num_heads 1 --sentencepiece True \
--steps 10000 --eval_steps 1000 --checkpoint 1000 --alpha 0.2 --dropout 0.2 \
--reg_scale 0.0 --decay True --decay_steps 5000 --lang eng --debug_mode False \
--eval 'data/processed_data/eng/eval_src' --eval_ref 'data/processed_data/eng/dev.reference0'