diff options
Diffstat (limited to 'tests/models/transformer/test_hard_aligns.sh')
-rw-r--r-- | tests/models/transformer/test_hard_aligns.sh | 14 |
1 files changed, 14 insertions, 0 deletions
diff --git a/tests/models/transformer/test_hard_aligns.sh b/tests/models/transformer/test_hard_aligns.sh new file mode 100644 index 0000000..6f5d82b --- /dev/null +++ b/tests/models/transformer/test_hard_aligns.sh @@ -0,0 +1,14 @@ +#!/bin/bash + +# Exit on error +set -e + +rm -f hardalign.out + +# Run Marian +$MRT_MARIAN/build/marian-decoder -c $MRT_MODELS/transformer/decode.yml -b 6 --mini-batch 32 --alignment < text.in > hardalign.out + +diff hardalign.out text.b6.hardalign.expected > hardalign.diff + +# Exit with success code +exit 0 |