Better fix for re-enabling training

tesseract-ocr · May 8, 2017 · b86b4fa · b86b4fa
1 parent 0afd593
commit b86b4fa
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/lstm/fullyconnected.cpp b/lstm/fullyconnected.cpp
@@ -65,7 +65,7 @@ void FullyConnected::SetEnableTraining(TrainingState state) {
     // Temp disable only from enabled.
     if (training_ == TS_ENABLED) training_ = state;
   } else {
-    if (state == TS_ENABLED && training_ == TS_DISABLED)
+    if (state == TS_ENABLED && training_ != TS_ENABLED)
       weights_.InitBackward();
     training_ = state;
   }

diff --git a/lstm/lstm.cpp b/lstm/lstm.cpp
@@ -113,7 +113,7 @@ void LSTM::SetEnableTraining(TrainingState state) {
     // Temp disable only from enabled.
     if (training_ == TS_ENABLED) training_ = state;
   } else {
-    if (state == TS_ENABLED && training_ == TS_DISABLED) {
+    if (state == TS_ENABLED && training_ != TS_ENABLED) {
       for (int w = 0; w < WT_COUNT; ++w) {
         if (w == GFS && !Is2D()) continue;
         gate_weights_[w].InitBackward();