diff options
Diffstat (limited to 'tests/interface/input-tsv/test_tsv_train_mini_batch_fit_stdin.sh')
-rw-r--r-- | tests/interface/input-tsv/test_tsv_train_mini_batch_fit_stdin.sh | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tests/interface/input-tsv/test_tsv_train_mini_batch_fit_stdin.sh b/tests/interface/input-tsv/test_tsv_train_mini_batch_fit_stdin.sh index 8b26362..0b4ba42 100644 --- a/tests/interface/input-tsv/test_tsv_train_mini_batch_fit_stdin.sh +++ b/tests/interface/input-tsv/test_tsv_train_mini_batch_fit_stdin.sh @@ -14,7 +14,7 @@ mkdir -p train_fit_stdin # Run marian command cat train.tsv | $MRT_MARIAN/marian \ - --no-shuffle --mini-batch-fit -w 500 --seed 2222 --dim-emb 32 --dim-rnn 64 --maxi-batch 10 --optimizer sgd \ + --cost-type ce-mean --no-shuffle --mini-batch-fit -w 500 --seed 2222 --dim-emb 32 --dim-rnn 64 --maxi-batch 10 --optimizer sgd \ -m train_fit_stdin/model.npz --tsv -t stdin -v $MRT_MODELS/rnn-spm/vocab.deen.{spm,spm} \ --disp-freq 4 --log train_fit_stdin.log |