if len(node_input_names) == 0:
if len(node_inputs) > 0:
if node_inputs[0] in model_inputs:
node_input_names.append(model_inputs[node_inputs[0]])
else:
input_name = "input{0}".format(input_index)
if input_name not in layers:
After Change
print("Graph outputs:", graph_outputs)
graph_inputs = [n.uniqueName() for n in trace.graph().inputs()]
print("Graph inputs:", graph_inputs)
// Collect model state dict