diff options
Diffstat (limited to 'tests/training/multi-gpu/test_async_sgd_runs.sh')
-rw-r--r-- | tests/training/multi-gpu/test_async_sgd_runs.sh | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tests/training/multi-gpu/test_async_sgd_runs.sh b/tests/training/multi-gpu/test_async_sgd_runs.sh index 1094cd2..5043548 100644 --- a/tests/training/multi-gpu/test_async_sgd_runs.sh +++ b/tests/training/multi-gpu/test_async_sgd_runs.sh @@ -13,7 +13,7 @@ rm -rf async_sgd async_sgd.log mkdir -p async_sgd $MRT_MARIAN/marian \ - --no-shuffle --seed 1111 --maxi-batch 1 --maxi-batch-sort none --dim-emb 128 --dim-rnn 256 \ + --no-shuffle --seed 1111 --maxi-batch 1 --maxi-batch-sort none --dim-emb 128 --dim-rnn 256 --cost-type ce-mean \ --devices 0 1 \ -m async_sgd/model.npz -t $MRT_DATA/europarl.de-en/corpus.bpe.{en,de} -v vocab.en.yml vocab.de.yml \ --disp-freq 20 --after-batches 400 \ |