修改bert工程格式

Python/bert.py

@@ -36,15 +36,8 @@ maxInput={"unique_ids_raw_output___9:0":[1], "input_ids:0":[1,256], "input_mask:
 model = migraphx.parse_onnx("../Resource/bertsquad-10.onnx", map_input_dims=maxInput)
 
 # 获取模型输入/输出节点信息
-print("inputs:")
 inputs = model.get_inputs()
-for key,value in inputs.items():
-    print("{}:{}".format(key,value))
-
-print("outputs:")
 outputs = model.get_outputs()
-for key,value in outputs.items():
-    print("{}:{}".format(key,value))
 
 # 编译
 model.compile(migraphx.get_target("gpu"),device_id=0)                        

Src/Bert.cpp

@@ -36,7 +36,7 @@ ErrorCode Bert::Initialize()
     onnx_options.map_input_dims["segment_ids:0"]={1,256};
 
     // 加载模型
-    if(Exists(modelPath)==false)
+    if(!Exists(modelPath))
     {
         LOG_ERROR(stdout,"%s not exist!\n",modelPath.c_str());
         return MODEL_NOT_EXIST;
@@ -45,18 +45,8 @@ ErrorCode Bert::Initialize()
     LOG_INFO(stdout,"succeed to load model: %s\n",GetFileName(modelPath).c_str());
 
     // 获取模型输入/输出节点信息
-    std::cout<<"inputs:"<<std::endl;
     std::unordered_map<std::string, migraphx::shape> inputs=net.get_inputs();
-    for(auto i:inputs)
-    {
-        std::cout<<i.first<<":"<<i.second<<std::endl;
-    }
-    std::cout<<"outputs:"<<std::endl;
     std::unordered_map<std::string, migraphx::shape> outputs=net.get_outputs();
-    for(auto i:outputs)
-    {
-        std::cout<<i.first<<":"<<i.second<<std::endl;
-    }
 
     inputName1 = "unique_ids_raw_output___9:0";
     inputShape1 = inputs.at(inputName1);