diff options
author | Roman Grundkiewicz <rgrundkiewicz@gmail.com> | 2020-10-21 13:24:14 +0300 |
---|---|---|
committer | Roman Grundkiewicz <rgrundkiewicz@gmail.com> | 2020-10-21 13:24:14 +0300 |
commit | 9e9c0e7620fb95576bd73afa6886d4059b9da0f4 (patch) | |
tree | 5dafa781ec7243e297234aea6a10697ec8571b52 /tests | |
parent | e7efe2bc2af3c52942e80921c2402b4c941011fb (diff) |
Update tests for cost-functions
Diffstat (limited to 'tests')
-rw-r--r-- | tests/training/cost-functions/test_ce-mean.sh | 1 | ||||
-rw-r--r-- | tests/training/cost-functions/test_ce-sum.sh | 2 |
2 files changed, 2 insertions, 1 deletions
diff --git a/tests/training/cost-functions/test_ce-mean.sh b/tests/training/cost-functions/test_ce-mean.sh index 9d49c48..c4109b1 100644 --- a/tests/training/cost-functions/test_ce-mean.sh +++ b/tests/training/cost-functions/test_ce-mean.sh @@ -14,6 +14,7 @@ rm -rf ce-mean ce-mean.log mkdir -p ce-mean $MRT_MARIAN/marian \ + --cost-type ce-mean \ --seed 9999 --sync-sgd \ -m ce-mean/model.npz -t $MRT_DATA/train.max50.{en,de} -v vocab.en.yml vocab.de.yml \ --disp-freq 2 --after-epochs 1 \ diff --git a/tests/training/cost-functions/test_ce-sum.sh b/tests/training/cost-functions/test_ce-sum.sh index dc8bfe0..f22f137 100644 --- a/tests/training/cost-functions/test_ce-sum.sh +++ b/tests/training/cost-functions/test_ce-sum.sh @@ -14,7 +14,7 @@ rm -rf ce-sum ce-sum.log mkdir -p ce-sum $MRT_MARIAN/marian \ - --cost-type ce-sum \ + --cost-type ce-sum --disp-label-counts false \ --seed 9999 --optimizer sgd --sync-sgd \ -m ce-sum/model.npz -t $MRT_DATA/train.max50.{en,de} -v vocab.en.yml vocab.de.yml \ --disp-freq 2 --after-epochs 1 \ |