diff options
Diffstat (limited to 'tests/interface/input-tsv/test_tsv_train_shuffle_in_ram.sh')
-rw-r--r-- | tests/interface/input-tsv/test_tsv_train_shuffle_in_ram.sh | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tests/interface/input-tsv/test_tsv_train_shuffle_in_ram.sh b/tests/interface/input-tsv/test_tsv_train_shuffle_in_ram.sh index d6c20e7..8ce67c3 100644 --- a/tests/interface/input-tsv/test_tsv_train_shuffle_in_ram.sh +++ b/tests/interface/input-tsv/test_tsv_train_shuffle_in_ram.sh @@ -14,7 +14,7 @@ mkdir -p train_shuffle_ram # Run marian command $MRT_MARIAN/marian \ - --shuffle-in-ram --seed 1111 --dim-emb 32 --dim-rnn 64 --maxi-batch 10 --optimizer sgd \ + --cost-type ce-mean --shuffle-in-ram --seed 1111 --dim-emb 32 --dim-rnn 64 --maxi-batch 10 --optimizer sgd \ -m train_shuffle_ram/model.npz --tsv --tsv-fields 2 -t train.tsv -v $MRT_MODELS/rnn-spm/vocab.deen.{spm,spm} \ --after-batches 20 --disp-freq 4 \ --log train_shuffle_ram.log |