1 files changed, 8 insertions, 10 deletions
diff --git a/tests/training/restoring/validation/test_restoring_stalled_validators.sh b/tests/training/restoring/validation/test_restoring_stalled_validators.sh
index 358c272..50a6d2e 100644
--- a/tests/training/restoring/validation/test_restoring_stalled_validators.sh
+++ b/tests/training/restoring/validation/test_restoring_stalled_validators.sh
@@ -7,16 +7,18 @@ set -e
 rm -rf valid_stalled valid_stalled_?.*log valid_script_?.temp
 mkdir -p valid_stalled
 
+extra_opts="--no-shuffle --seed 2222 --maxi-batch 1 --maxi-batch-sort none --quiet-translation"
+extra_opts="$extra_opts --dim-emb 64 --dim-rnn 128 --mini-batch 16 --optimizer sgd"
+extra_opts="$extra_opts --cost-type ce-mean --disp-label-counts false"
+
+
 head -n 8 $MRT_DATA/europarl.de-en/toy.bpe.en > valid.mini.bpe.en
 head -n 8 $MRT_DATA/europarl.de-en/toy.bpe.de > valid.mini.bpe.de
 
-
 # Uncomment to re-generate the expected output
 #mkdir -p valid_stalled_full
 
-#$MRT_MARIAN/marian \
-    #--no-shuffle --seed 2222 --maxi-batch 1 --maxi-batch-sort none --quiet-translation \
-    #--dim-emb 64 --dim-rnn 128 --mini-batch 16 --optimizer sgd \
+#$MRT_MARIAN/marian $extra_opts \
     #-m valid_stalled_full/model.npz -t $MRT_DATA/europarl.de-en/toy.bpe.{en,de} -v vocab.en.yml vocab.de.yml \
     #--disp-freq 10 --valid-freq 20 --after-batches 200 --early-stopping 5 \
     #--valid-metrics cross-entropy valid-script translation --valid-script-path ./valid_script_ab.sh \
@@ -33,9 +35,7 @@ head -n 8 $MRT_DATA/europarl.de-en/toy.bpe.de > valid.mini.bpe.de
 #exit 1
 
 
-$MRT_MARIAN/marian \
-    --no-shuffle --seed 2222 --maxi-batch 1 --maxi-batch-sort none --quiet-translation \
-    --dim-emb 64 --dim-rnn 128 --mini-batch 16 --optimizer sgd \
+$MRT_MARIAN/marian $extra_opts \
     -m valid_stalled/model.npz -t $MRT_DATA/europarl.de-en/toy.bpe.{en,de} -v vocab.en.yml vocab.de.yml \
     --disp-freq 10 --valid-freq 20 --after-batches 100 --early-stopping 5 \
     --valid-metrics cross-entropy valid-script translation --valid-script-path ./valid_script_ab.sh \
@@ -53,9 +53,7 @@ cat valid_stalled_1.log | $MRT_TOOLS/strip-timestamps.sh \
     | head -n -1 > valid_stalled.out
 
 
-$MRT_MARIAN/marian \
-    --no-shuffle --seed 2222 --maxi-batch 1 --maxi-batch-sort none --quiet-translation \
-    --dim-emb 64 --dim-rnn 128 --mini-batch 16 --optimizer sgd \
+$MRT_MARIAN/marian $extra_opts \
     -m valid_stalled/model.npz -t $MRT_DATA/europarl.de-en/toy.bpe.{en,de} -v vocab.en.yml vocab.de.yml \
     --disp-freq 10 --valid-freq 20 --after-batches 200 --early-stopping 5 \
     --valid-metrics cross-entropy valid-script translation --valid-script-path ./valid_script_ab.sh \