diff options
Diffstat (limited to 'tests/training/restoring/validation/test_valid_reset_stalled.sh')
| -rw-r--r-- | tests/training/restoring/validation/test_valid_reset_stalled.sh | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/tests/training/restoring/validation/test_valid_reset_stalled.sh b/tests/training/restoring/validation/test_valid_reset_stalled.sh index 02b24a5..3e96636 100644 --- a/tests/training/restoring/validation/test_valid_reset_stalled.sh +++ b/tests/training/restoring/validation/test_valid_reset_stalled.sh @@ -16,11 +16,13 @@ mkdir -p valid_reset_stalled test -s valid.mini.bpe.en || head -n 8 $MRT_DATA/europarl.de-en/toy.bpe.en > valid.mini.bpe.en test -s valid.mini.bpe.de || head -n 8 $MRT_DATA/europarl.de-en/toy.bpe.de > valid.mini.bpe.de +extra_opts="--no-shuffle --seed 222 --maxi-batch 1 --maxi-batch-sort none" +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" + # Train a model for a while and stop -$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_reset_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 140 --early-stopping 5 \ --valid-metrics translation valid-script cross-entropy --valid-script-path ./valid_script_ab.sh \ @@ -36,9 +38,7 @@ cat valid_reset_stalled_1.log | $MRT_TOOLS/strip-timestamps.sh | grep -P "\[vali # Restart training with --valid-reset-stalled -$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_reset_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-reset-stalled \ --valid-metrics translation valid-script cross-entropy --valid-script-path ./valid_script_ab.sh \ |