1 files changed, 7 insertions, 6 deletions
diff --git a/tests/training/restoring/validation/test_restoring_validation_lower_is_better.sh b/tests/training/restoring/validation/test_restoring_validation_lower_is_better.sh
index 33ad8d0..a29e534 100644
--- a/tests/training/restoring/validation/test_restoring_validation_lower_is_better.sh
+++ b/tests/training/restoring/validation/test_restoring_validation_lower_is_better.sh
@@ -7,10 +7,13 @@ set -e
 rm -rf valid_lowisbet valid_lowisbet_?.log
 mkdir -p valid_lowisbet
 
+extra_opts="--no-shuffle --seed 1111 --maxi-batch 1 --maxi-batch-sort none"
+extra_opts="$extra_opts --dim-emb 64 --dim-rnn 128 --mini-batch 32"
+extra_opts="$extra_opts --cost-type ce-mean --disp-label-counts false"
+
+
 # Files for the validation sets are swapped intentionally
-$MRT_MARIAN/marian \
-    --no-shuffle --seed 1111 --maxi-batch 1 --maxi-batch-sort none \
-    --dim-emb 64 --dim-rnn 128 --mini-batch 32 \
+$MRT_MARIAN/marian $extra_opts \
     -m valid_lowisbet/model.npz -t $MRT_DATA/train.max50.{en,de} -v vocab.en.yml vocab.de.yml \
     --disp-freq 10 --valid-freq 30 --after-batches 160 --early-stopping 2 \
     --valid-metrics cross-entropy --valid-sets $MRT_DATA/europarl.de-en/toy.bpe.{de,en} --valid-mini-batch 64 \
@@ -24,9 +27,7 @@ cp valid_lowisbet/model.npz.progress.yml valid_lowisbet/model.npz.progress.yml.b
 cat valid_lowisbet_1.log | $MRT_TOOLS/strip-timestamps.sh | grep "cross-entropy" > valid_lowisbet.out
 
 # Files for the validation sets are swapped intentionally
-$MRT_MARIAN/marian \
-    --no-shuffle --seed 1111 --maxi-batch 1 --maxi-batch-sort none \
-    --dim-emb 64 --dim-rnn 128 --mini-batch 32 \
+$MRT_MARIAN/marian $extra_opts \
     -m valid_lowisbet/model.npz -t $MRT_DATA/train.max50.{en,de} -v vocab.en.yml vocab.de.yml \
     --disp-freq 10 --valid-freq 30 --after-batches 320 --early-stopping 4 \
     --valid-metrics cross-entropy --valid-sets $MRT_DATA/europarl.de-en/toy.bpe.{de,en} --valid-mini-batch 64 \