-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathjob_glove.sh
executable file
·54 lines (48 loc) · 1.51 KB
/
job_glove.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
46
47
48
49
50
51
52
#!/bin/bash
if [ ! -d "/s0/ttmt001" ]
then
space_req s0
echo "Creating scratch space"
fi
source /g/ssli/transitory/ttmt001/envs/py3.6-torch1.7-gpu/bin/activate
#source /g/ssli/transitory/ttmt001/envs/py3.6-torch1.7-cpu/bin/activate
for history_len in 1 3
do
for gen_type in beam greedy
do
python train_baseline.py \
--attr_embedding_dim 30 \
--sent_encoder_hidden_dim 100 \
--n_sent_encoder_layers 1 \
--dial_encoder_hidden_dim 200 \
--n_dial_encoder_layers 1 \
--decoder_hidden_dim 200 \
--n_decoder_layers 1 \
--batch_size 64 --eval_batch_size 64 \
--rnn_type gru \
--history_len ${history_len} \
--gen_type ${gen_type} \
--top_k 10 \
--filename_note 1x3x-glove-${gen_type}-hist${history_len}
done
done
for history_len in 1 3
do
for gen_type in beam greedy
do
python train_baseline.py \
--attr_embedding_dim 64 \
--sent_encoder_hidden_dim 128 \
--n_sent_encoder_layers 2 \
--dial_encoder_hidden_dim 256 \
--n_dial_encoder_layers 2 \
--decoder_hidden_dim 256 \
--n_decoder_layers 2 \
--batch_size 64 --eval_batch_size 64 \
--rnn_type gru \
--history_len ${history_len} \
--gen_type ${gen_type} \
--top_k 10 \
--filename_note 5x7x-glove-${gen_type}-hist${history_len}
done
done