diff --git a/src/net.cpp b/src/net.cpp index b68334463..db6fb6f92 100644 --- a/src/net.cpp +++ b/src/net.cpp @@ -633,6 +633,13 @@ int Net::load_model(FILE* fp) for (size_t i=0; iload_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) {