Unverified Commit 5a0f39b5 by pyjhzwh Committed by GitHub

[Torch] fix unordered dictionary problem for python version under 3.6 (#4982)

* fix unordered dictionary problem for python version 3.5

* modify style
parent 98b17590
...@@ -918,7 +918,7 @@ def _get_constant(node): ...@@ -918,7 +918,7 @@ def _get_constant(node):
def _get_operator_nodes(nodes): def _get_operator_nodes(nodes):
""" Returns torch IR nodes that need conversion to Relay """ """ Returns torch IR nodes that need conversion to Relay """
ops = {} ops = []
# Traverse nodes and add to graph # Traverse nodes and add to graph
for node in nodes: for node in nodes:
if node.outputsSize() > 1: if node.outputsSize() > 1:
...@@ -927,7 +927,7 @@ def _get_operator_nodes(nodes): ...@@ -927,7 +927,7 @@ def _get_operator_nodes(nodes):
node_name = _get_output_name(node) node_name = _get_output_name(node)
if node.kind() != "prim::GetAttr": if node.kind() != "prim::GetAttr":
ops[node_name] = node ops.append((node_name, node))
return ops return ops
...@@ -1015,7 +1015,7 @@ def parse_params(graph, state_dict): ...@@ -1015,7 +1015,7 @@ def parse_params(graph, state_dict):
def parse_operators(operators, outputs, output_index_map, ret_name): def parse_operators(operators, outputs, output_index_map, ret_name):
""" Convert each Torch IR operators to Relay equivalent """ """ Convert each Torch IR operators to Relay equivalent """
for node_name, op_node in operators.items(): for node_name, op_node in operators:
operator = op_node.kind() operator = op_node.kind()
inputs = _get_op_inputs(op_node, outputs, output_index_map) inputs = _get_op_inputs(op_node, outputs, output_index_map)
......
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