diff options
author | Roman Grundkiewicz <rgrundkiewicz@gmail.com> | 2020-10-21 13:46:37 +0300 |
---|---|---|
committer | Roman Grundkiewicz <rgrundkiewicz@gmail.com> | 2020-10-21 13:46:37 +0300 |
commit | b7f466c361d593a247fc866ad25ef4458cfb597e (patch) | |
tree | b3b89bfbb9c518fc75a996fb182125fa7afd17d6 /tests/training/models/lm | |
parent | 4abbfe39580f520ae0d7659c996d251c025876c4 (diff) |
Update tests for different models
Diffstat (limited to 'tests/training/models/lm')
-rw-r--r-- | tests/training/models/lm/test_lm-transformer.sh | 2 | ||||
-rw-r--r-- | tests/training/models/lm/test_lm.sh | 2 |
2 files changed, 2 insertions, 2 deletions
diff --git a/tests/training/models/lm/test_lm-transformer.sh b/tests/training/models/lm/test_lm-transformer.sh index 916d08e..476c2e2 100644 --- a/tests/training/models/lm/test_lm-transformer.sh +++ b/tests/training/models/lm/test_lm-transformer.sh @@ -15,7 +15,7 @@ mkdir -p lm-transformer $MRT_MARIAN/marian \ --seed 1111 --no-shuffle \ - --type lm-transformer --dim-emb 128 --dim-rnn 256 \ + --type lm-transformer --dim-emb 128 --dim-rnn 256 --cost-type ce-mean \ -m lm-transformer/model.npz -t $MRT_DATA/europarl.de-en/corpus.bpe.en -v vocab.en.yml \ --disp-freq 20 --after-batches 100 \ --log lm-transformer.log 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 |