21994919156aac15558f77555538346fb702bcbc,autokeras/nn/generator.py,ResNetGenerator,_make_block,#ResNetGenerator#Any#Any#Any#Any#Any#,189

Before Change

        out = graph.add_layer(self.batch_norm(planes), out)
        if downsample is not None:
            downsample_out = graph.add_layer(StubReLU(), node_id)
            downsample_out = graph.add_layer(downsample[0], downsample_out)
            residual_node_id = graph.add_layer(downsample[1], downsample_out)
        out = graph.add_layer(StubAdd(), (out, residual_node_id))
        return out

After Change

        out = graph.add_layer(self.conv(planes, planes, kernel_size=3), out)

        residual_node_id = graph.add_layer(StubReLU(), residual_node_id)
        residual_node_id = graph.add_layer(self.conv(in_planes,
                                                     planes * self.block_expansion,
                                                     kernel_size=1,
                                                     stride=stride), residual_node_id)
        out = graph.add_layer(StubAdd(), (out, residual_node_id))
        return out

Instances