b0cd6a1df09c7157dea48265a5c21c3b7cc37399,softlearning/misc/nn.py,,feedforward_model,#Any#Any#Any#Any#Any#Any#,4

Before Change

                      name=None,
                      *args,
                      **kwargs):
    if not isinstance(input_shapes[0], (list, tuple)):
        raise NotImplementedError(
            "TODO(hartikainen): feedforward_model currently expects a list of"
            " shapes as an input. It might be possible that you passed in a"

After Change

        model.add(tf.keras.layers.Dense(
            units, *args, activation=activation, **kwargs))

    model.add(tf.keras.layers.Dense(
        output_size, *args, activation=output_activation, **kwargs))

    return model

Instances