Bug fix.

zhongkaifu · Sep 1, 2023 · 716bef2 · 716bef2
1 parent 4159ef9
commit 716bef2
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 6 deletions.
diff --git a/Seq2SeqSharp/Corpus/MonoCorpus.cs b/Seq2SeqSharp/Corpus/MonoCorpus.cs
@@ -344,10 +344,13 @@ public IEnumerator<T> GetEnumerator()
                         {
                             var tgtLine = tgtLines[i];
 
-                            if ((100 * batchIdx / m_batchNumInTotal) > currentBatchPercent)
+                            if (m_batchNumInTotal > 0)
                             {
-                                Logger.WriteLine($"Processing batch '{batchIdx}/{m_batchNumInTotal}'."); // The '{i}th' record in this batch is: Target = '{tgtLine}'");
-                                currentBatchPercent++;
+                                if ((100 * batchIdx / m_batchNumInTotal) > currentBatchPercent)
+                                {
+                                    Logger.WriteLine($"Processing batch '{batchIdx}/{m_batchNumInTotal}'."); // The '{i}th' record in this batch is: Target = '{tgtLine}'");
+                                    currentBatchPercent++;
+                                }
                             }
 
                             SntPair sntPair = new SntPair(tgtLine, tgtLine);

diff --git a/Seq2SeqSharp/Corpus/ParallelCorpus.cs b/Seq2SeqSharp/Corpus/ParallelCorpus.cs
@@ -442,10 +442,13 @@ public IEnumerator<T> GetEnumerator()
                             var srcLine = srcLines[i];
                             var tgtLine = tgtLines[i];
 
-                            if ((100 * batchIdx / m_batchNumInTotal) > currentBatchPercent)
+                            if (m_batchNumInTotal > 0)
                             {
-                                Logger.WriteLine($"Processing batch '{batchIdx}/{m_batchNumInTotal}'."); // The '{i}th' record in this batch is: Source = '{srcLine}' Target = '{tgtLine}'");
-                                currentBatchPercent++;
+                                if ((100 * batchIdx / m_batchNumInTotal) > currentBatchPercent)
+                                {
+                                    Logger.WriteLine($"Processing batch '{batchIdx}/{m_batchNumInTotal}'."); // The '{i}th' record in this batch is: Source = '{srcLine}' Target = '{tgtLine}'");
+                                    currentBatchPercent++;
+                                }
                             }
 
                             SntPair sntPair = new SntPair(srcLine, tgtLine);