def generate_model_config(name,  n_filters, nonlinearity='relu'):
    """ Generates a spec describing how to build the model

    Args:
      name: name for this model config
      n_filters: list of num filters for each layer - [n_layers]

    Returns:
      model_spec
    """
    model_config = {
        'name': name,
        'n_filters': n_filters,
        'nonlinearity': nonlinearity
    }
    return model_config