Browse Source

fix(lite/load_and_run): fix invalid model type caused by oss model magic change

GitOrigin-RevId: 1d071f999e
revert-411-add-tools
Megvii Engine Team 3 years ago
parent
commit
9646f1aac4
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      lite/load_and_run/src/models/model.cpp

+ 1
- 1
lite/load_and_run/src/models/model.cpp View File

@@ -24,12 +24,12 @@ ModelType ModelBase::get_model_type(std::string model_path) {
fclose(fin);

// get model type
// uint32_t MGB_MAGIC = 0x5342474D
std::string tag(buf);
ModelType type;
if (tag.substr(0, 7) == std::string("mgb0001") ||
tag.substr(0, 8) == std::string("mgb0000a") ||
tag.substr(0, 4) == std::string("MGBS") ||
tag.substr(0, 4) == std::string("MGBC") ||
tag.substr(0, 8) == std::string("mgbtest0")) {
type = ModelType::MEGDL_MODEL;



Loading…
Cancel
Save