diff options
Diffstat (limited to 'tests/training/restoring/corpus/test_corpus_restoration_one_epoch.sh')
-rw-r--r-- | tests/training/restoring/corpus/test_corpus_restoration_one_epoch.sh | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tests/training/restoring/corpus/test_corpus_restoration_one_epoch.sh b/tests/training/restoring/corpus/test_corpus_restoration_one_epoch.sh index 1006398..345c7af 100644 --- a/tests/training/restoring/corpus/test_corpus_restoration_one_epoch.sh +++ b/tests/training/restoring/corpus/test_corpus_restoration_one_epoch.sh @@ -10,7 +10,7 @@ mkdir -p corpus_one test -e vocab.de.yml test -e vocab.en.yml -extra_opts="--seed 9999 --maxi-batch 1 --maxi-batch-sort none --mini-batch 32 -o sgd --dim-emb 128 --dim-rnn 256 --disp-freq 4" +extra_opts="--seed 9999 --maxi-batch 1 --maxi-batch-sort none --mini-batch 32 --optimizer sgd --dim-emb 128 --dim-rnn 256 --disp-freq 4" $MRT_MARIAN/build/marian \ -m corpus_one/model_full.npz -t $MRT_DATA/train.max50.{en,de} -v vocab.en.yml vocab.de.yml \ |