def splitter(model):
    """
    Splits model parameters into multiple groups to allow fine-tuning
    """
    params = list(model.parameters())
    return [
        # weights and biases of the first linear layer
        params[:2],
        # weights and biases of the second linear layer
        params[2:],
    ]