def evaluate_mutable_model_fun(model, variables, σ, mutable):
    """
    Default function to evaluate the model during training, with
    mutable argument passed so that mutated state can be passed back
    """
    return model.apply(variables, σ, mutable=mutable)