diff options
Diffstat (limited to 'tests/training/cost-functions/test_ce-mean.sh')
-rw-r--r-- | tests/training/cost-functions/test_ce-mean.sh | 1 |
1 files changed, 1 insertions, 0 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 \ |