Update tests for different models

7 files changed, 7 insertions, 7 deletions

diff --git a/tests/training/models/lm/test_lm-transformer.sh b/tests/training/models/lm/test_lm-transformer.sh
index 916d08e..476c2e2 100644
--- a/tests/training/models/lm/test_lm-transformer.sh
+++ b/tests/training/models/lm/test_lm-transformer.sh
@@ -15,7 +15,7 @@ mkdir -p lm-transformer
 
 $MRT_MARIAN/marian \
     --seed 1111 --no-shuffle \
-    --type lm-transformer --dim-emb 128 --dim-rnn 256 \
+    --type lm-transformer --dim-emb 128 --dim-rnn 256 --cost-type ce-mean \
     -m lm-transformer/model.npz -t $MRT_DATA/europarl.de-en/corpus.bpe.en -v vocab.en.yml \
     --disp-freq 20 --after-batches 100 \
     --log lm-transformer.log
diff --git a/tests/training/models/lm/test_lm.sh b/tests/training/models/lm/test_lm.sh
index 75503a6..91f94d4 100644
--- a/tests/training/models/lm/test_lm.sh
+++ b/tests/training/models/lm/test_lm.sh
@@ -15,7 +15,7 @@ mkdir -p lm
 
 $MRT_MARIAN/marian \
     --seed 1111 --no-shuffle \
-    --type lm --dim-emb 128 --dim-rnn 256 \
+    --type lm --dim-emb 128 --dim-rnn 256 --cost-type ce-mean \
     -m lm/model.npz -t $MRT_DATA/europarl.de-en/corpus.bpe.en -v vocab.en.yml \
     --disp-freq 20 --after-batches 100 \
     --log lm.log
diff --git a/tests/training/models/multi-source/test_multi-s2s.sh b/tests/training/models/multi-source/test_multi-s2s.sh
index e53808e..a1ceef1 100644
--- a/tests/training/models/multi-source/test_multi-s2s.sh
+++ b/tests/training/models/multi-source/test_multi-s2s.sh
@@ -15,7 +15,7 @@ mkdir -p multi-s2s
 
 $MRT_MARIAN/marian \
     --seed 1111 --no-shuffle \
-    --type multi-s2s --dim-emb 128 --dim-rnn 256 \
+    --type multi-s2s --dim-emb 128 --dim-rnn 256 --cost-type ce-mean \
     -m multi-s2s/model.npz -t train.bpe.{en,xx,de} -v vocab.en.yml vocab.xx.yml vocab.de.yml \
     --disp-freq 20 --after-batches 100 \
     --log multi-s2s.log
diff --git a/tests/training/models/multi-source/test_multi-transformer.sh b/tests/training/models/multi-source/test_multi-transformer.sh
index b263f98..425ebdc 100644
--- a/tests/training/models/multi-source/test_multi-transformer.sh
+++ b/tests/training/models/multi-source/test_multi-transformer.sh
@@ -15,7 +15,7 @@ mkdir -p multi-transformer
 
 $MRT_MARIAN/marian \
     --seed 1111 --no-shuffle \
-    --type multi-transformer --dim-emb 128 --dim-rnn 256 \
+    --type multi-transformer --dim-emb 128 --dim-rnn 256 --cost-type ce-mean \
     -m multi-transformer/model.npz -t train.bpe.{en,xx,de} -v vocab.en.yml vocab.xx.yml vocab.de.yml \
     --disp-freq 20 --after-batches 100 \
     --log multi-transformer.log
diff --git a/tests/training/models/nematus/test_encdec_depth.sh b/tests/training/models/nematus/test_encdec_depth.sh
index fd08c9d..ed5276e 100644
--- a/tests/training/models/nematus/test_encdec_depth.sh
+++ b/tests/training/models/nematus/test_encdec_depth.sh
@@ -17,7 +17,7 @@ $MRT_MARIAN/marian \
     --type nematus --enc-cell gru-nematus --dec-cell gru-nematus \
     --enc-depth 4 --enc-cell-depth 4 --enc-type bidirectional --dec-depth 4 --dec-cell-base-depth 4 --dec-cell-high-depth 1 \
     --layer-normalization \
-    --no-shuffle --seed 1111 --dim-emb 64 --dim-rnn 128 \
+    --no-shuffle --seed 1111 --dim-emb 64 --dim-rnn 128 --cost-type ce-mean \
     -m encdec_depth/model.npz -t $MRT_DATA/europarl.de-en/corpus.bpe.{de,en} -v vocab.en.yml vocab.de.yml \
     --log encdec_depth.log --disp-freq 2 --after-batches 10
 
diff --git a/tests/training/models/nematus/test_wmt17_model.sh b/tests/training/models/nematus/test_wmt17_model.sh
index 5c06c83..d43ff9c 100644
--- a/tests/training/models/nematus/test_wmt17_model.sh
+++ b/tests/training/models/nematus/test_wmt17_model.sh
@@ -11,7 +11,7 @@ $MRT_MARIAN/marian \
     --type nematus --enc-cell gru-nematus --dec-cell gru-nematus \
     --enc-depth 1 --enc-cell-depth 4 --enc-type bidirectional --dec-depth 1 --dec-cell-base-depth 8 --dec-cell-high-depth 1 \
     --layer-normalization \
-    --no-shuffle --seed 1111 --dim-emb 64 --dim-rnn 128 \
+    --no-shuffle --seed 1111 --dim-emb 64 --dim-rnn 128 --cost-type ce-mean \
     -m wmt17/model.npz -t $MRT_DATA/europarl.de-en/corpus.bpe.{de,en} -v vocab.en.yml vocab.de.yml \
     --log wmt17.log --disp-freq 2 --after-batches 10
 
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 \