99773d6efd4c081424bdbc3ee9871cdf3e1bdb5c,autokeras/graph.py,Graph,to_add_skip_model,#Graph#Any#Any#,239

Before Change

        layer = WeightedAdd()
        single_input_shape = get_int_tuple(start.output_shape)
        layer.build([single_input_shape, single_input_shape])
        self._add_edge(layer, new_node_id, self.adj_list[output_id][][0], False)
        self._add_edge(layer, input_id, self.adj_list[output_id][0][0], False)

        self._redirect_edge(output_id, self.adj_list[output_id][0][0], new_node_id)

After Change

        layer.build([single_input_shape, single_input_shape])

        relu_output_id = self.adj_list[relu_input_id][0][0]
        self._redirect_edge(relu_input_id, relu_output_id, new_node_id)
        self._add_edge(layer, new_node_id, relu_output_id, False)
        self._add_edge(layer, skip_output_id, relu_output_id, False)

        return self.produce_model()

Instances