outputs = []
input_index = 0
model_inputs = dict()
for node in nodes:
node_inputs = list(node.inputs())
node_input_names = []
for node_input in node_inputs:
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
state_dict = _unique_state_dict(model)