mirror/tesseract
1
0
Fork 0
mirror of https://github.com/tesseract-ocr/tesseract.git synced 2024-11-24 02:59:07 +08:00
Code Issues Actions 10 Packages Projects Releases Wiki Activity

Abort LSTM training with integer model (fixes issue #1573)

Browse Source 
Tesseract currently cannot continue LSTM training from an
integer (fast) model.

Report this to users who try it nevertheless instead of crashing
with an assertion.

Signed-off-by: Stefan Weil <sw@weilnetz.de>
This commit is contained in:
Stefan Weil 2021-09-06 08:17:18 +02:00
parent ffb71e5793
commit 15aee49d0a
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/lstm/lstmtrainer.cpp
View File

@ -132,6 +132,10 @@ bool LSTMTrainer::TryLoadingCheckpoint(const char* filename,
if (!(*file_reader_)(filename, &data)) return false;
tprintf("Loaded file %s, unpacking...\n", filename);
if (!checkpoint_reader_->Run(data, this)) return false;
if (IsIntMode()) {
tprintf("Error, %s is an integer (fast) model, cannot continue training\n", filename);
return false;
}
StaticShape shape = network_->OutputShape(network_->InputShape());
if (((old_traineddata == nullptr || *old_traineddata == '\0') &&
network_->NumOutputs() == recoder_.code_range()) ||