Browse Source

Fix crash issue during load_model (#848)

* Fix crash issue during load_model

* Fix crash issue during load_model 2nd part
tags/20190611
Gemfield nihui 7 years ago
parent
commit
573c2bcd93
1 changed files with 13 additions and 0 deletions
  1. +13
    -0
      src/net.cpp

+ 13
- 0
src/net.cpp View File

@@ -633,6 +633,13 @@ int Net::load_model(FILE* fp)
for (size_t i=0; i<layers.size(); i++)
{
Layer* layer = layers[i];
//Here we found inconsistent content in the parameter file.
if (!layer){
fprintf(stderr, "load_model error at layer %d, parameter file has inconsistent content.\n", (int)i);
ret = -1;
break;
}

int lret = layer->load_model(mb);
if (lret != 0)
@@ -823,6 +830,12 @@ int Net::load_model(const unsigned char* _mem)
{
Layer* layer = layers[i];

//Here we found inconsistent content in the parameter file.
if (!layer){
fprintf(stderr, "load_model error at layer %d, parameter file has inconsistent content.\n", (int)i);
return -1;
}

int lret = layer->load_model(mb);
if (lret != 0)
{


Loading…
Cancel
Save