diff options
Diffstat (limited to 'tests/interface/input-tsv/test_tsv_train_with_align_shuffle_in_ram.sh')
-rw-r--r-- | tests/interface/input-tsv/test_tsv_train_with_align_shuffle_in_ram.sh | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/tests/interface/input-tsv/test_tsv_train_with_align_shuffle_in_ram.sh b/tests/interface/input-tsv/test_tsv_train_with_align_shuffle_in_ram.sh index 49e74de..a5db769 100644 --- a/tests/interface/input-tsv/test_tsv_train_with_align_shuffle_in_ram.sh +++ b/tests/interface/input-tsv/test_tsv_train_with_align_shuffle_in_ram.sh @@ -14,10 +14,11 @@ mkdir -p train_align_shuffle_ram # Run marian command $MRT_MARIAN/marian \ - --cost-type ce-mean --shuffle-in-ram --clip-norm 1 --seed 4444 --dim-emb 32 --dim-rnn 64 --maxi-batch 1 --maxi-batch-sort none --optimizer sgd --learn-rate 0.1 \ + --cost-type ce-mean --shuffle-in-ram --clip-norm 1 --seed 4444 --maxi-batch 1 --maxi-batch-sort none --optimizer adam --learn-rate 0.1 \ + --dim-emb 32 --transformer-dim-ffn 64 --type transformer --enc-depth 3 --dec-depth 3 \ -m train_align_shuffle_ram/model.npz --tsv -t train2.aln-de-en.tsv -v $MRT_MODELS/rnn-spm/vocab.deen.{spm,spm} \ --after-batches 100 --disp-freq 4 \ - --guided-alignment 0 --guided-alignment-weight 1.0 \ + --guided-alignment 0 --guided-alignment-weight 1.0 --guided-alignment-cost ce \ --log train_align_shuffle_ram.log # Check if files exist |