Browse Source

!908 dts:display model info

From: @wangwenhua1
Reviewed-by: @xchu42,@ji_chen
Signed-off-by: @ji_chen
tags/v1.2.0
mindspore-ci-bot Gitee 3 years ago
parent
commit
f7f47c90d4
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)";

static const char *const kCaffeFormatSupport = "only support NCHW, ND in Caffe model";


Loading…
Cancel
Save