def packnet_simple_mlp(
    num_classes=10,
    input_size=28 * 28,
    hidden_size=512,
    hidden_layers=1,
    drop_rate=0.5,
) -> PackNetModel:
    """
    Convenience function for creating a PackNet compatible :class:`SimpleMLP`
    model.

    :param num_classes: output size
    :param input_size: input size
    :param hidden_size: hidden layer size
    :param hidden_layers: number of hidden layers
    :param drop_rate: dropout rate. 0 to disable
    :return: A PackNet compatible model
    """
    return PackNetModel(
        SimpleMLP(num_classes, input_size, hidden_size, hidden_layers, drop_rate)
    )

    Convenience function for creating a PackNet compatible :class:`SimpleMLP`
    model.

    :param num_classes: output size
    :param input_size: input size
    :param hidden_size: hidden layer size
    :param hidden_layers: number of hidden layers
    :param drop_rate: dropout rate. 0 to disable
    :return: A PackNet compatible model
    
avalanche/models/packnet.py