diff options
| author | Marcin Junczys-Dowmunt <junczys@amu.edu.pl> | 2017-04-09 20:08:42 +0300 |
|---|---|---|
| committer | Marcin Junczys-Dowmunt <junczys@amu.edu.pl> | 2017-04-09 20:08:42 +0300 |
| commit | fecea28165fa4db98010542e90ce3cb93dc257b4 (patch) | |
| tree | e32e14e48b8d2c1b7f868d2ad121ae3eba97b679 | |
| parent | bb5b353c5af059bf6cde6ea5f41361aef62cb015 (diff) | |
modify training example to use dynamic batching
| -rwxr-xr-x | examples/training/run-me.sh | 8 | ||||
| m--------- | src/marian | 11 |
2 files changed, 9 insertions, 10 deletions
diff --git a/examples/training/run-me.sh b/examples/training/run-me.sh index 605ad302..fc489c2a 100755 --- a/examples/training/run-me.sh +++ b/examples/training/run-me.sh @@ -48,20 +48,20 @@ fi if [ ! -e "model/model.npz" ] then -../../build/marian \ +../../../marian/build/marian \ --model model/model.npz \ --devices $GPUS --seed 0 \ --train-sets data/corpus.bpe.ro data/corpus.bpe.en \ --vocabs model/vocab.ro.yml model/vocab.en.yml \ --dim-vocabs 66000 50000 \ - --mini-batch 80 \ + --dynamic-batching -w 3000 \ --layer-normalization --dropout-rnn 0.2 --dropout-src 0.1 --dropout-trg 0.1 \ --early-stopping 5 --moving-average \ - --valid-freq 10000 --save-freq 10000 --disp-freq 1000 \ + --valid-freq 10000 --save-freq 10000 --disp-freq 100 \ --valid-sets data/newsdev2016.bpe.ro data/newsdev2016.bpe.en \ --valid-metrics cross-entropy valid-script \ --valid-script-path ./scripts/validate.sh \ - --log model/train.log --valid-log model/valid.log + --log model/train.log --valid-log model/valid.log fi diff --git a/src/marian b/src/marian -Subproject 85cee316ff3cb8d4ef53a208de8d63168b9406f +Subproject 6b801f9964b7c30496d4c86befabf238c23d5cd |