b048efa4f956b80266942caea6ee8b4311e1d17a,autokeras/net_transformer.py,,to_wider_graph,#Any#,9

Before Change

        The wider model
    
    // The last conv layer cannot be widen since wider operator cannot be done over the two sides of flatten.
    conv_layers = list(filter(lambda x: is_conv_layer(x), graph.layer_list))[:-1]
    // The first layer cannot be widen since widen operator cannot be done over the two sides of flatten.
    // The last layer is softmax, which also cannot be widen.
    dense_layers = list(filter(lambda x: is_dense_layer(x), graph.layer_list))[1:-1]

    if len(dense_layers) == 0:
        weighted_layers = conv_layers
    elif randint(0, 1) == 0:
        weighted_layers = conv_layers
    else:
        weighted_layers = dense_layers

    if len(weighted_layers) <= 1:

After Change

    if is_conv_layer(graph.layer_list[target_id]):
        n_add = randint(1, 4 * graph.layer_list[target_id].filters)
    else:
        n_add = randint(1, 4 * graph.layer_list[target_id].units)

    graph.to_wider_model(target_id, n_add)
    return graph

Instances