Merge pull request #72 from marian-nmt/pr-mjd-mbfstep

Update regression tests for new mini-batch-fit step size

9 files changed, 43 insertions, 29 deletions

diff --git a/tests/interface/input-tsv/train_fit.expected b/tests/interface/input-tsv/train_fit.expected
index 9788052..1c2d2eb 100644
--- a/tests/interface/input-tsv/train_fit.expected
+++ b/tests/interface/input-tsv/train_fit.expected
@@ -1,5 +1,5 @@
-281.93161011
-221.91622925
-252.93228149
-227.49211121
-277.62591553
+251.22476196
+252.45635986
+252.93251038
+258.67086792
+232.73229980
diff --git a/tests/interface/input-tsv/train_fit_stdin.expected b/tests/interface/input-tsv/train_fit_stdin.expected
index 71223fc..b9ce675 100644
--- a/tests/interface/input-tsv/train_fit_stdin.expected
+++ b/tests/interface/input-tsv/train_fit_stdin.expected
@@ -1,3 +1,3 @@
-352.16757202
-180.04814148
-248.64807129
+344.08602905
+173.01716614
+248.64839172
diff --git a/tests/training/basics/batch_fit.expected b/tests/training/basics/batch_fit.expected
index 3769b33..b52e2e2 100644
--- a/tests/training/basics/batch_fit.expected
+++ b/tests/training/basics/batch_fit.expected
@@ -1,5 +1,5 @@
 320
-316
-310
-312
-312
+320
+319
+320
+320
diff --git a/tests/training/basics/test_mini_batch_fit.sh b/tests/training/basics/test_mini_batch_fit.sh
index 37b7389..4f57c11 100644
--- a/tests/training/basics/test_mini_batch_fit.sh
+++ b/tests/training/basics/test_mini_batch_fit.sh
@@ -3,7 +3,7 @@
 #####################################################################
 # SUMMARY: Train with --mini-batch-fit
 # AUTHOR: snukky
-# TAGS: batchfit
+# TAGS: mini-batch-fit
 #####################################################################
 
 # Exit on error
diff --git a/tests/training/features/data-weighting/sqlite_word.expected b/tests/training/features/data-weighting/sqlite_word.expected
index 037b678..fb557e0 100644
--- a/tests/training/features/data-weighting/sqlite_word.expected
+++ b/tests/training/features/data-weighting/sqlite_word.expected
@@ -1,15 +1,14 @@
-862.09051514
-728.33001709
-630.45318604
-554.12792969
-469.74276733
-387.57510376
-299.93112183
-257.35052490
-828.63208008
-706.60028076
-615.24633789
-537.03155518
-454.32131958
-369.54190063
-274.37387085
+853.40081787
+710.09143066
+610.75262451
+526.56585693
+439.88232422
+345.80377197
+234.99189758
+606.08709717
+734.31378174
+627.71948242
+544.36926270
+456.04959106
+364.86871338
+264.01800537
diff --git a/tests/training/features/data-weighting/test_word_weighting_sqlite.sh b/tests/training/features/data-weighting/test_word_weighting_sqlite.sh
index 1658e1b..fe399c7 100644
--- a/tests/training/features/data-weighting/test_word_weighting_sqlite.sh
+++ b/tests/training/features/data-weighting/test_word_weighting_sqlite.sh
@@ -2,7 +2,7 @@
 
 #####################################################################
 # SUMMARY:
-# TAGS: dataweights
+# TAGS: dataweights mini-batch-fit
 #####################################################################
 
 # Exit on error
diff --git a/tests/training/restoring/corpus/test_corpus_restoration_mini_batch_fit.sh b/tests/training/restoring/corpus/test_corpus_restoration_mini_batch_fit.sh
index f32aac2..ad544e5 100644
--- a/tests/training/restoring/corpus/test_corpus_restoration_mini_batch_fit.sh
+++ b/tests/training/restoring/corpus/test_corpus_restoration_mini_batch_fit.sh
@@ -1,5 +1,10 @@
 #!/bin/bash -x
 
+#####################################################################
+# SUMMARY:
+# TAGS: mini-batch-fit
+#####################################################################
+
 # Exit on error
 set -e
 
diff --git a/tests/training/restoring/corpus/test_corpus_restoration_mini_batch_fit_sync.sh b/tests/training/restoring/corpus/test_corpus_restoration_mini_batch_fit_sync.sh
index f1b1815..7e8dc5e 100644
--- a/tests/training/restoring/corpus/test_corpus_restoration_mini_batch_fit_sync.sh
+++ b/tests/training/restoring/corpus/test_corpus_restoration_mini_batch_fit_sync.sh
@@ -1,5 +1,10 @@
 #!/bin/bash -x
 
+#####################################################################
+# SUMMARY:
+# TAGS: mini-batch-fit
+#####################################################################
+
 # Exit on error
 set -e
 
diff --git a/tests/training/restoring/corpus/test_corpus_restoration_no_shuffle.sh b/tests/training/restoring/corpus/test_corpus_restoration_no_shuffle.sh
index 7b6da5e..d147702 100644
--- a/tests/training/restoring/corpus/test_corpus_restoration_no_shuffle.sh
+++ b/tests/training/restoring/corpus/test_corpus_restoration_no_shuffle.sh
@@ -1,5 +1,10 @@
 #!/bin/bash -x
 
+#####################################################################
+# SUMMARY:
+# TAGS: mini-batch-fit
+#####################################################################
+
 # Exit on error
 set -e