diff options
Diffstat (limited to 'tests/training/features/guided-alignment/test_guided_alignment_transformer.sh')
-rw-r--r-- | tests/training/features/guided-alignment/test_guided_alignment_transformer.sh | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tests/training/features/guided-alignment/test_guided_alignment_transformer.sh b/tests/training/features/guided-alignment/test_guided_alignment_transformer.sh index d678a7a..f5f18b9 100644 --- a/tests/training/features/guided-alignment/test_guided_alignment_transformer.sh +++ b/tests/training/features/guided-alignment/test_guided_alignment_transformer.sh @@ -3,6 +3,7 @@ ##################################################################### # SUMMARY: Training transformer model with guided alignment # AUTHOR: snukky +# TAGS: align transformer ##################################################################### # Exit on error @@ -14,7 +15,7 @@ mkdir -p transformer # Run marian command $MRT_MARIAN/marian --type transformer \ - --no-shuffle --seed 2222 --dim-emb 32 --dim-rnn 64 --mini-batch 32 --maxi-batch 1 --maxi-batch-sort none --optimizer sgd \ + --no-shuffle --seed 2222 --dim-emb 32 --dim-rnn 64 --mini-batch 32 --maxi-batch 1 --maxi-batch-sort none --optimizer sgd --cost-type ce-mean \ -m transformer/model.npz -t corpus.bpe.{en,de} -v vocab.en.yml vocab.de.yml \ --after-batches 100 --disp-freq 10 \ --guided-alignment corpus.bpe.align --guided-alignment-weight 1.0 --learn-rate 0.1 \ |