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;
model->fbmodel = tflite::FlatBufferModel::BuildFromFile(aModelPath);
if (status != kTfLiteOk) {
std::cerr << status << std::endl;
return status;
if (!model->fbmodel) {
std::cerr << "Error at reading model file " << aModelPath << std::endl;
return kTfLiteError;
}
tflite::ops::builtin::BuiltinOpResolver resolver;
status = tflite::InterpreterBuilder(*model->fbmodel, resolver)(&model->interpreter);
if (status != kTfLiteOk) {
std::cerr << status << std::endl;
return status;
tflite::InterpreterBuilder(*model->fbmodel, resolver)(&model->interpreter);
if (!model->interpreter) {
std::cerr << "Error at InterpreterBuilder for model file " << aModelPath << std::endl;
return kTfLiteError;
}
model->interpreter->AllocateTensors();