-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathrun.small-gpt2.sh
executable file
·34 lines (26 loc) · 1.37 KB
/
run.small-gpt2.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
for mode in "penman-var" "penman-frames-var"
do
IFS=","
for seed in "run-1",13 "run-2",14 "run-3",15 "run-4",16
do
read -a run <<< "$seed"
python Train.py -train-src data/amr/2/$mode/train/amr.txt \
-train-tgt data/amr/2/$mode/train/sentence.txt \
-dev-src data/amr/2/$mode/dev/amr.txt -dev-tgt data/amr/2/$mode/dev/sentence.txt \
-test-src data/amr/2/$mode/test/amr.txt \
-batch-size 4 -src-max-length 180 -tgt-max-length 180 -lr 5e-4 \
-epochs 12 -gpu -print-every 10 -model pierreguillou/gpt2-small-portuguese \
-save-dir output/gpt2-small/2/$mode/fixed/${run[0]} -seed ${run[1]} -beam-size 15 \
-representation amr -fixed-embed -pretrained-model gpt2 -accum-steps 8 \
-early-stopping-patience 4 -early-stopping-criteria perplexity
python Train.py -train-src data/amr/2/$mode/train/amr.txt \
-train-tgt data/amr/2/$mode/train/sentence.txt \
-dev-src data/amr/2/$mode/dev/amr.txt -dev-tgt data/amr/2/$mode/dev/sentence.txt \
-test-src data/amr/2/$mode/test/amr.txt \
-batch-size 4 -src-max-length 180 -tgt-max-length 180 -lr 5e-4 \
-epochs 12 -gpu -print-every 10 -model pierreguillou/gpt2-small-portuguese \
-save-dir output/gpt2-small/2/$mode/nofixed/${run[0]} -seed ${run[1]} -beam-size 15 \
-representation amr -pretrained-model gpt2 -accum-steps 8 \
-early-stopping-patience 4 -early-stopping-criteria perplexity
done
done