Browse Source

display model info

tags/v1.2.0
wangwenhua1@huawei.com 4 years ago
parent
commit
c6aa3841d9
1 changed files with 2 additions and 1 deletions
  1. +2
    -1
      ge/offline/main.cc

+ 2
- 1
ge/offline/main.cc View File

@@ -65,7 +65,8 @@ using std::vector;
static bool is_dynamic_input = false;

const char *const kModeSupport = "only support 0(model to framework model), "
"1(framework model to json), 3(only pre-check), 5(pbtxt to json)";
"1(framework model to json), 3(only pre-check), "
"5(pbtxt to json), 6(display model info)";
const char *const kModelToJsonSupport = "only support 0(Caffe) 3(TensorFlow) 5(Onnx)";

// limit available mem size 2G


Loading…
Cancel
Save