experiments/STT
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Proper TFLite error checking

Browse Source 
Fixes #1842
This commit is contained in:
Alexandre Lissy 2019-01-22 15:53:33 +01:00
parent 092ad2773c
commit 9b845533c9
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
native_client/deepspeech.cc
View File

@ -579,17 +579,17 @@ DS_CreateModel(const char* aModelPath,
TfLiteStatus status; TfLiteStatus status;
model->fbmodel = tflite::FlatBufferModel::BuildFromFile(aModelPath); model->fbmodel = tflite::FlatBufferModel::BuildFromFile(aModelPath);
if (status != kTfLiteOk) { if (!model->fbmodel) {
std::cerr << status << std::endl; std::cerr << "Error at reading model file " << aModelPath << std::endl;
return status; return kTfLiteError;
} }
tflite::ops::builtin::BuiltinOpResolver resolver; tflite::ops::builtin::BuiltinOpResolver resolver;
status = tflite::InterpreterBuilder(*model->fbmodel, resolver)(&model->interpreter); tflite::InterpreterBuilder(*model->fbmodel, resolver)(&model->interpreter);
if (status != kTfLiteOk) { if (!model->interpreter) {
std::cerr << status << std::endl; std::cerr << "Error at InterpreterBuilder for model file " << aModelPath << std::endl;
return status; return kTfLiteError;
} }
model->interpreter->AllocateTensors(); model->interpreter->AllocateTensors();