diff options
Diffstat (limited to 'tests/interface/input-tsv/test_tsv_train_with_align_and_weights_inputtypes.sh')
-rw-r--r-- | tests/interface/input-tsv/test_tsv_train_with_align_and_weights_inputtypes.sh | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tests/interface/input-tsv/test_tsv_train_with_align_and_weights_inputtypes.sh b/tests/interface/input-tsv/test_tsv_train_with_align_and_weights_inputtypes.sh index a983f22..8653a67 100644 --- a/tests/interface/input-tsv/test_tsv_train_with_align_and_weights_inputtypes.sh +++ b/tests/interface/input-tsv/test_tsv_train_with_align_and_weights_inputtypes.sh @@ -14,7 +14,7 @@ mkdir -p train_align_weights_intypes # Run marian command $MRT_MARIAN/marian \ - --no-shuffle --seed 7777 --dim-emb 32 --dim-rnn 64 --maxi-batch 1 --maxi-batch-sort none --optimizer sgd --learn-rate 0.1 \ + --cost-type ce-mean --no-shuffle --seed 7777 --dim-emb 32 --dim-rnn 64 --maxi-batch 1 --maxi-batch-sort none --optimizer sgd --learn-rate 0.1 \ -m train_align_weights_intypes/model.npz --tsv -t train2.de-w-aln-en.tsv -v $MRT_MODELS/rnn-spm/vocab.deen.{spm,spm} \ --after-batches 60 --disp-freq 4 \ --input-types sequence weight alignment sequence --guided-alignment-weight 1.0 \ |