Merge pull request tesseract-ocr#1493 from stweil/train

lstmtraining: Fix handling of --max_iterations
zwlijun · Apr 19, 2018 · a474fd0 · a474fd0
2 parents 36d6cb3 + f95041f
commit a474fd0
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/training/lstmtraining.cpp b/training/lstmtraining.cpp
@@ -195,7 +195,8 @@ int main(int argc, char **argv) {
     // Train a few.
     int iteration = trainer.training_iteration();
     for (int target_iteration = iteration + kNumPagesPerBatch;
-         iteration < target_iteration;
+         iteration < target_iteration &&
+         (iteration < FLAGS_max_iterations || FLAGS_max_iterations == 0);
          iteration = trainer.training_iteration()) {
       trainer.TrainOnLine(&trainer, false);
     }