From c6aa3841d9f3109c9232e5ea677d4967eabc4f2c Mon Sep 17 00:00:00 2001 From: "wangwenhua1@huawei.com" Date: Fri, 8 Jan 2021 15:45:20 +0800 Subject: [PATCH] display model info --- ge/offline/main.cc | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/ge/offline/main.cc b/ge/offline/main.cc index 5b3eb4ae..f1c38517 100755 --- a/ge/offline/main.cc +++ b/ge/offline/main.cc @@ -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