diff options
| author | Roman Grundkiewicz <rgrundkiewicz@gmail.com> | 2020-10-21 15:14:19 +0300 |
|---|---|---|
| committer | Roman Grundkiewicz <rgrundkiewicz@gmail.com> | 2020-10-21 15:14:19 +0300 |
| commit | ca4b0504acc2bab8f013145ceffd9c87d60d077a (patch) | |
| tree | 71abb21a17a530047cde97601dd7568558921ec0 /tests | |
| parent | 1d37c13ac10e3c75dc979d96bdb432f1a2d4a6d6 (diff) | |
Update tests for restarting optimizers
Diffstat (limited to 'tests')
5 files changed, 6 insertions, 4 deletions
diff --git a/tests/training/restoring/optimizer/test_adagrad_params.sh b/tests/training/restoring/optimizer/test_adagrad_params.sh index 4b5d420..8fca356 100644 --- a/tests/training/restoring/optimizer/test_adagrad_params.sh +++ b/tests/training/restoring/optimizer/test_adagrad_params.sh @@ -10,7 +10,7 @@ mkdir -p adagrad $MRT_MARIAN/marian \ --no-shuffle --seed 7777 --maxi-batch 1 --maxi-batch-sort none --dim-emb 128 --dim-rnn 256 \ -m adagrad/model.npz -t $MRT_DATA/europarl.de-en/corpus.bpe.{en,de} -v vocab.en.yml vocab.de.yml \ - --disp-freq 10 --after-batches 100 --save-freq 60 --optimizer adagrad \ + --disp-freq 10 --after-batches 100 --save-freq 60 --optimizer adagrad --cost-type ce-mean \ --log adagrad.log test -e adagrad/model.npz diff --git a/tests/training/restoring/optimizer/test_adam_params.sh b/tests/training/restoring/optimizer/test_adam_params.sh index e70663f..e0bd76f 100644 --- a/tests/training/restoring/optimizer/test_adam_params.sh +++ b/tests/training/restoring/optimizer/test_adam_params.sh @@ -10,7 +10,7 @@ mkdir -p adam $MRT_MARIAN/marian \ --no-shuffle --seed 7777 --maxi-batch 1 --maxi-batch-sort none --dim-emb 128 --dim-rnn 256 \ -m adam/model.npz -t $MRT_DATA/europarl.de-en/corpus.bpe.{en,de} -v vocab.en.yml vocab.de.yml \ - --disp-freq 10 --after-batches 100 --save-freq 60 \ + --disp-freq 10 --after-batches 100 --save-freq 60 --cost-type ce-mean \ --log adam.log test -e adam/model.npz diff --git a/tests/training/restoring/optimizer/test_adam_params_async.sh b/tests/training/restoring/optimizer/test_adam_params_async.sh index cb335c0..2b2c869 100644 --- a/tests/training/restoring/optimizer/test_adam_params_async.sh +++ b/tests/training/restoring/optimizer/test_adam_params_async.sh @@ -15,7 +15,7 @@ fi $MRT_MARIAN/marian \ --no-shuffle --seed 7777 --maxi-batch 1 --maxi-batch-sort none --mini-batch 32 --dim-emb 128 --dim-rnn 256 \ -m adam_async/model.npz -t $MRT_DATA/europarl.de-en/corpus.bpe.{en,de} -v vocab.en.yml vocab.de.yml \ - --disp-freq 10 --after-batches 100 --save-freq 60 --cost-type ce-sum \ + --disp-freq 10 --after-batches 100 --save-freq 60 --cost-type ce-sum --disp-label-counts false \ --log adam_async.log --devices 0 1 test -e adam_async/model.npz diff --git a/tests/training/restoring/optimizer/test_adam_params_sync.sh b/tests/training/restoring/optimizer/test_adam_params_sync.sh index 6697e9b..1e2481d 100644 --- a/tests/training/restoring/optimizer/test_adam_params_sync.sh +++ b/tests/training/restoring/optimizer/test_adam_params_sync.sh @@ -16,7 +16,7 @@ $MRT_MARIAN/marian \ --no-shuffle --seed 7777 --maxi-batch 1 --maxi-batch-sort none --mini-batch 32 --dim-emb 128 --dim-rnn 256 \ -m adam_sync/model.npz -t $MRT_DATA/europarl.de-en/corpus.bpe.{en,de} -v vocab.en.yml vocab.de.yml \ --disp-freq 10 --after-batches 100 --save-freq 60 \ - --log adam_sync.log --devices 0 1 --sync-sgd --cost-type ce-sum --clip-norm 0 + --log adam_sync.log --devices 0 1 --sync-sgd --cost-type ce-sum --disp-label-counts false --clip-norm 0 test -e adam_sync/model.npz test -e adam_sync/model.npz.optimizer.npz diff --git a/tests/training/restoring/optimizer/test_loading_adam_params.sh b/tests/training/restoring/optimizer/test_loading_adam_params.sh index 00c0cdc..95a48c8 100644 --- a/tests/training/restoring/optimizer/test_loading_adam_params.sh +++ b/tests/training/restoring/optimizer/test_loading_adam_params.sh @@ -8,6 +8,8 @@ rm -rf adam_load adam_load_?.log mkdir -p adam_load extra_opts="--no-shuffle --seed 7777 --maxi-batch 1 --maxi-batch-sort none --mini-batch 2 --dim-rnn 64 --dim-emb 32" +# Added because default options has changes +extra_opts="$extra_opts --cost-type ce-mean --disp-label-counts false" $MRT_MARIAN/marian \ -m adam_load/model.npz -t $MRT_DATA/train.max50.{en,de} -v vocab.en.yml vocab.de.yml \ |