diff options
Diffstat (limited to 'tests/training/models/lm/test_lm.sh')
-rw-r--r-- | tests/training/models/lm/test_lm.sh | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tests/training/models/lm/test_lm.sh b/tests/training/models/lm/test_lm.sh index 75503a6..91f94d4 100644 --- a/tests/training/models/lm/test_lm.sh +++ b/tests/training/models/lm/test_lm.sh @@ -15,7 +15,7 @@ mkdir -p lm $MRT_MARIAN/marian \ --seed 1111 --no-shuffle \ - --type lm --dim-emb 128 --dim-rnn 256 \ + --type lm --dim-emb 128 --dim-rnn 256 --cost-type ce-mean \ -m lm/model.npz -t $MRT_DATA/europarl.de-en/corpus.bpe.en -v vocab.en.yml \ --disp-freq 20 --after-batches 100 \ --log lm.log |