5101a0fe2a89683ea3df8cb896610aaa3da04fb1,pytorch2keras/converter.py,,pytorch_to_keras,#,141

Before Change

                s = i
            k += 1
            if k == len(seq_to_find):
                print("found seq", k, s)
                reshape_op = nodes[s + k - 1]
                flatten_op = {
                    "kind": (lambda: "onnx::Flatten"),
                    "attributeNames": (lambda: {}),

After Change

        node_id = get_node_id(node)
        node_name_regex = re.findall(r"\[([\w\d.\-\[\]\s]+)\]", node_scope_name)

        try: 
            int(node_name_regex[-1])
            node_weigth_group_name = ".".join(
                node_name_regex[:-1]
            )
            node_weights_name = node_weigth_group_name + "." + str(group_indices[node_weigth_group_name])
            group_indices[node_weigth_group_name] += 1

        except ValueError:
            node_weights_name = ".".join(
                node_name_regex
            )
        except IndexError:
            node_weights_name = ".".join(node_input_names)

        node_attrs = {k: node[k] for k in node.attributeNames()}

        node_outputs = list(node.outputs())
        node_outputs_names = []

Instances