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

LSTMTrainer: Catch empty vectors

Browse Source 
The new test in LSTMTrainer::UpdateErrorGraph fixes an assertion
(see issues #644, #792).

The new test in LSTMTrainer::ReadTrainingDump was added to improve
the robustness of the code.

Signed-off-by: Stefan Weil <sw@weilnetz.de>
This commit is contained in:
Stefan Weil 2017-06-04 17:41:20 +02:00 committed by Stefan Weil
parent 1e5522d321
commit 34d1e7331d
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
lstm/lstmtrainer.cpp
View File

@ -918,6 +918,10 @@ bool LSTMTrainer::SaveTrainingDump(SerializeAmount serialize_amount,
// Reads previously saved trainer from memory.
bool LSTMTrainer::ReadTrainingDump(const GenericVector<char>& data,
LSTMTrainer* trainer) {
if (data.size() == 0) {
tprintf("Warning: data size is zero in LSTMTrainer::ReadTrainingDump\n");
return false;
}
return trainer->ReadSizedTrainingDump(&data[0], data.size());
}
@ -1298,8 +1302,9 @@ STRING LSTMTrainer::UpdateErrorGraph(int iteration, double error_rate,
if (error_rate < best_error_rate_) {
// This is a new (global) minimum.
if (tester != NULL) {
result = tester->Run(worst_iteration_, worst_error_rates_,
worst_model_data_, CurrentTrainingStage());
if (worst_model_data_.size() != 0)
result = tester->Run(worst_iteration_, worst_error_rates_,
worst_model_data_, CurrentTrainingStage());
worst_model_data_.truncate(0);
best_model_data_ = model_data;
}