display model info

4 years ago · c6aa3841d9
parent 028df1048a
commit c6aa3841d9
1 changed files with 2 additions and 1 deletions
--- 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