blob: a4fa869150843f7bfc98896444f2d75715c3f510 (
plain)
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
53
54
55
56
57
58
59
60
61
62
63
64
|
#!/bin/bash -v
if [ ! -e ../../build/amun ]
then
echo "amun is not installed in ../../build, you need to compile the toolkit first."
exit 1
fi
if [ ! -e ../../build/marian ]
then
echo "marian is not installed in ../../build, you need to compile the toolkit first."
exit 1
fi
if [ ! -e "mosesdecoder" ]
then
git clone https://github.com/moses-smt/mosesdecoder
fi
if [ ! -e "subword-nmt" ]
then
git clone https://github.com/rsennrich/subword-nmt
fi
if [ ! -e "data/ro-en.tgz" ]
then
./scripts/download-files.sh
fi
mkdir model
if [ ! -e "data/corpus.bpe.en" ]
then
./scripts/preprocess.sh
fi
if [ ! -e "model/model.npz" ]
then
../../build/marian \
--model model/model.npz \
--devices 0 \
--train-sets data/corpus.bpe.ro data/corpus.bpe.en \
--vocabs model/vocab.ro.yml model/vocab.en.yml \
--dim-vocabs 66000 50000 \
--mini-batch 80 \
--layer-normalization \
--early-stopping 5 \
--valid-freq 10000 --save-freq 30000 --disp-freq 1000 \
--valid-sets data/newsdev2016.bpe.ro data/newsdev2016.bpe.en \
--valid-metrics cross-entropy valid-script \
--valid-script-path ./scripts/validate.sh \
--log model/train.log --valid-log model/valid.log
fi
if [ ! -e "data/newstest2016.bpe.ro.output.postprocessed" ]
then
cat data/newstest2016.bpe.ro \
| ../../build/amun -c model/model.npz.amun.yml -b 12 -n --mini-batch 100 --maxi-batch 1000 \
| sed 's/\@\@ //g' | mosesdecoder/scripts/recaser/detruecase.perl > data/newstest2016.bpe.ro.output.postprocessed
fi
./mosesdecoder/scripts/generic/multi-bleu.perl data/newstest2016.tok.en < data/newstest2016.bpe.ro.output.postprocessed
|