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

additional refinement for input and output name processing

parent f5409f95
......@@ -136,6 +136,7 @@ def get_input_shapes(sample_case, param_names):
for i in range(len(param_names)):
file_name = sample_case + '/input_' + str(i) + '.pb'
name, data = read_pb_file(file_name)
param_shape_map[name] = data.shape
shape_array.append(data.shape)
if name:
name_array.append(name)
......
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