diff options
Diffstat (limited to 'tests/training/features/data-weighting/test_word_weighting_sqlite.sh')
-rw-r--r-- | tests/training/features/data-weighting/test_word_weighting_sqlite.sh | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/tests/training/features/data-weighting/test_word_weighting_sqlite.sh b/tests/training/features/data-weighting/test_word_weighting_sqlite.sh index 229d5a8..fe399c7 100644 --- a/tests/training/features/data-weighting/test_word_weighting_sqlite.sh +++ b/tests/training/features/data-weighting/test_word_weighting_sqlite.sh @@ -1,5 +1,10 @@ #!/bin/bash +##################################################################### +# SUMMARY: +# TAGS: dataweights mini-batch-fit +##################################################################### + # Exit on error set -e @@ -10,7 +15,7 @@ mkdir -p sqlite_word cat $MRT_DATA/europarl.de-en/toy.bpe.en | sed -r 's/[^ ]+/2/g' > sqlite_word.weights.txt $MRT_MARIAN/marian \ - --seed 1111 --no-shuffle --dim-emb 128 --dim-rnn 256 --optimizer sgd \ + --seed 1111 --no-shuffle --dim-emb 128 --dim-rnn 256 --optimizer sgd --cost-type ce-mean \ -m sqlite_word/model.npz -t $MRT_DATA/europarl.de-en/toy.bpe.{de,en} -v vocab.{de,en}.yml \ --log sqlite_word.log --disp-freq 5 -e 2 --mini-batch-fit -w 500 \ --data-weighting sqlite_word.weights.txt --data-weighting-type word --sqlite sqlite_word/corpus.sqlite3 |