mirror/tesseract
1
0
Fork 0
mirror of https://github.com/tesseract-ocr/tesseract.git synced 2025-01-18 06:30:14 +08:00
Code Issues Actions 10 Packages Projects Releases Wiki Activity

Better fix for re-enabling training

Browse Source
This commit is contained in:
Ray Smith 2017-05-08 14:26:09 -07:00
parent 0afd5939b1
commit b86b4fa06b
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lstm/fullyconnected.cpp
View File

@ -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;
}

2
lstm/lstm.cpp
View File

@ -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();