Commit df2da5de authored by Shucai Xiao's avatar Shucai Xiao
Browse files

additional refinement of print out info

parent 6323fc2c
...@@ -221,7 +221,6 @@ def main(): ...@@ -221,7 +221,6 @@ def main():
# get param names # get param names
param_names = model_parameter_names(model_path_name) param_names = model_parameter_names(model_path_name)
print("param_name = {}".format(param_names))
# get output names # get output names
output_names = model_output_names(model_path_name) output_names = model_output_names(model_path_name)
...@@ -232,6 +231,7 @@ def main(): ...@@ -232,6 +231,7 @@ def main():
param_shapes = get_input_shapes(sample_case, param_names) param_shapes = get_input_shapes(sample_case, param_names)
for name, dims in param_shapes.items(): for name, dims in param_shapes.items():
print("Input: {}, shape: {}".format(name, dims)) print("Input: {}, shape: {}".format(name, dims))
print()
# read and compile model # read and compile model
model = migraphx.parse_onnx(model_path_name, map_input_dims=param_shapes) model = migraphx.parse_onnx(model_path_name, map_input_dims=param_shapes)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment