diff options
Diffstat (limited to 'tests/training/multi-gpu/test_sync_sgd.sh')
-rw-r--r-- | tests/training/multi-gpu/test_sync_sgd.sh | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tests/training/multi-gpu/test_sync_sgd.sh b/tests/training/multi-gpu/test_sync_sgd.sh index ac2d009..ca29645 100644 --- a/tests/training/multi-gpu/test_sync_sgd.sh +++ b/tests/training/multi-gpu/test_sync_sgd.sh @@ -15,7 +15,7 @@ mkdir -p sync_sgd $MRT_MARIAN/marian \ --no-shuffle --seed 777 --mini-batch 4 --maxi-batch 1 --maxi-batch-sort none \ --dim-rnn 64 --dim-emb 32 --learn-rate 0.001 --clip-norm 0 \ - --devices 0 1 --sync-sgd --optimizer sgd \ + --devices 0 1 --sync-sgd --optimizer sgd --cost-type ce-mean \ -m sync_sgd/model.full.npz -t $MRT_DATA/europarl.de-en/corpus.bpe.{en,de} -v vocab.en.yml vocab.de.yml \ --disp-freq 10 --after-batches 100 \ --log sync_sgd.log |