diff options
Diffstat (limited to 'tests/training/models/transformer/test_transformer.sh')
-rw-r--r-- | tests/training/models/transformer/test_transformer.sh | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tests/training/models/transformer/test_transformer.sh b/tests/training/models/transformer/test_transformer.sh index e9754cf..b1ad881 100644 --- a/tests/training/models/transformer/test_transformer.sh +++ b/tests/training/models/transformer/test_transformer.sh @@ -7,7 +7,7 @@ set -e 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" +opts="--no-shuffle --seed 1111 --mini-batch 32 --maxi-batch 1 --maxi-batch-sort none --optimizer sgd --dim-emb 64 --dim-rnn 128 --cost-type ce-mean" $MRT_MARIAN/marian \ --type transformer -m transformer/model.npz -t $MRT_DATA/europarl.de-en/corpus.bpe.{en,de} -v vocab.en.yml vocab.de.yml \ |