diff options
author | Roman Grundkiewicz <rgrundki@exseed.ed.ac.uk> | 2018-11-28 19:58:01 +0300 |
---|---|---|
committer | Roman Grundkiewicz <rgrundki@exseed.ed.ac.uk> | 2018-11-28 19:58:01 +0300 |
commit | 8b50eb0ae106bee2fa58a8005c1819c85d00bda3 (patch) | |
tree | 6420b72307cc6fd44b321cd5676dbbf2073ca2e1 /tests/training/models/transformer | |
parent | 9cc169495729ce9d25ffae7492f08fd6d8619915 (diff) |
Organize tests
Diffstat (limited to 'tests/training/models/transformer')
-rw-r--r-- | tests/training/models/transformer/.gitignore | 2 | ||||
-rw-r--r-- | tests/training/models/transformer/test_transformer.sh | 24 | ||||
-rw-r--r-- | tests/training/models/transformer/transformer.expected | 10 |
3 files changed, 36 insertions, 0 deletions
diff --git a/tests/training/models/transformer/.gitignore b/tests/training/models/transformer/.gitignore new file mode 100644 index 0000000..9a5cf30 --- /dev/null +++ b/tests/training/models/transformer/.gitignore @@ -0,0 +1,2 @@ +transformer +decoder_c_tt diff --git a/tests/training/models/transformer/test_transformer.sh b/tests/training/models/transformer/test_transformer.sh new file mode 100644 index 0000000..35f8f09 --- /dev/null +++ b/tests/training/models/transformer/test_transformer.sh @@ -0,0 +1,24 @@ +#!/bin/bash -x + +# Exit on error +set -e + +# Test code goes here +rm -rf transformer transformer*.log +mkdir -p transformer + +opts="--no-shuffle --seed 1111 --mini-batch 32 --maxi-batch 1 --maxi-batch-sort none --optimizer sgd --dim-emb 64 --dim-rnn 128" + +$MRT_MARIAN/build/marian \ + --type transformer -m transformer/model.npz -t $MRT_DATA/europarl.de-en/corpus.bpe.{en,de} -v vocab.en.yml vocab.de.yml \ + $opts --disp-freq 10 --after-batches 100 \ + --log transformer.log + +test -e transformer/model.npz +test -e transformer.log + +cat transformer.log | $MRT_TOOLS/extract-costs.sh > transformer.out +$MRT_TOOLS/diff-nums.py transformer.out transformer.expected -p 0.01 -o transformer.diff + +# Exit with success code +exit 0 diff --git a/tests/training/models/transformer/transformer.expected b/tests/training/models/transformer/transformer.expected new file mode 100644 index 0000000..5bc6a78 --- /dev/null +++ b/tests/training/models/transformer/transformer.expected @@ -0,0 +1,10 @@ +226.90 +250.85 +244.19 +247.59 +242.19 +238.91 +236.10 +231.21 +237.86 +241.79 |