diff options
Diffstat (limited to 'tests/training/restarting/test_sgd_for_two_epochs.sh')
-rw-r--r-- | tests/training/restarting/test_sgd_for_two_epochs.sh | 9 |
1 files changed, 8 insertions, 1 deletions
diff --git a/tests/training/restarting/test_sgd_for_two_epochs.sh b/tests/training/restarting/test_sgd_for_two_epochs.sh index ad92b5f..11bf76e 100644 --- a/tests/training/restarting/test_sgd_for_two_epochs.sh +++ b/tests/training/restarting/test_sgd_for_two_epochs.sh @@ -1,5 +1,11 @@ #!/bin/bash -x +##################################################################### +# SUMMARY: Restaring training after the 1st epoch (async) +# AUTHOR: snukky +# TAGS: optimizer clip-norm +##################################################################### + # Exit on error set -e @@ -7,12 +13,13 @@ set -e rm -rf sgd_2e sgd_1st_epoch.log sgd_2nd_epoch.log mkdir -p sgd_2e -extra_opts="--no-shuffle --seed 1111 --maxi-batch 1 --maxi-batch-sort none --mini-batch 32 --optimizer sgd" +extra_opts="--no-shuffle --clip-norm 1 --seed 1111 --maxi-batch 1 --maxi-batch-sort none --mini-batch 32 --optimizer sgd" # Added because default options has changes extra_opts="$extra_opts --cost-type ce-mean --disp-label-counts false" # Uncomment to prepare the expected output +#rm -f sgd_two_epochs.log #$MRT_MARIAN/marian \ #-m sgd_2e/model_2e.npz -t $MRT_DATA/train.max50.{en,de} -v vocab.en.yml vocab.de.yml \ #--disp-freq 4 --save-freq 32 --after-epochs 2 -l 0.1 $extra_opts \ |