mirror/tesseract
1
0
Fork 0
mirror of https://github.com/tesseract-ocr/tesseract.git synced 2025-06-11 12:43:17 +08:00
Code Issues Actions 10 Packages Projects Releases Wiki Activity

Merge pull request #978 from stweil/lstm

Browse Source 
LSTMTrainer: Catch empty vectors
This commit is contained in:
zdenop 2017-06-12 19:32:45 +02:00 committed by GitHub
commit 2b854e3749
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. // Reads previously saved trainer from memory.
bool LSTMTrainer::ReadTrainingDump(const GenericVector<char>& data, bool LSTMTrainer::ReadTrainingDump(const GenericVector<char>& data,
LSTMTrainer* trainer) { 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()); 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_) { if (error_rate < best_error_rate_) {
// This is a new (global) minimum. // This is a new (global) minimum.
if (tester != NULL) { if (tester != NULL) {
result = tester->Run(worst_iteration_, worst_error_rates_, if (worst_model_data_.size() != 0)
worst_model_data_, CurrentTrainingStage()); result = tester->Run(worst_iteration_, worst_error_rates_,
worst_model_data_, CurrentTrainingStage());
worst_model_data_.truncate(0); worst_model_data_.truncate(0);
best_model_data_ = model_data; best_model_data_ = model_data;
} }