Browse Source

display model info

pull/688/head
wangwenhua1@huawei.com 5 years ago
parent
commit
4dad1b8467
1 changed files with 5 additions and 1 deletions
  1. +5
    -1
      ge/common/helper/model_helper.cc

+ 5
- 1
ge/common/helper/model_helper.cc View File

@@ -129,7 +129,11 @@ Status ModelHelper::SaveModelDef(std::shared_ptr<OmFileSaveHelper> &om_file_save
model_tmp->SetGraph(ge_model->GetGraph());
model_tmp->SetVersion(ge_model->GetVersion());
model_tmp->SetAttr(ge_model->MutableAttrMap());
SaveSizeToModelDef(ge_model);
Status ret = SaveSizeToModelDef(ge_model);
if (ret != SUCCESS) {
GELOGE(ret, "SaveSizeToModelDef failed");
return ret;
}

(void)model_tmp->Save(model_buffer);
GELOGD("MODEL_DEF size is %zu", model_buffer.GetSize());


Loading…
Cancel
Save