Learn R Programming

ggmlR (version 0.6.1)

ggml_unfreeze_weights: Unfreeze Layer Weights

Description

Sets trainable = TRUE on layers. Accepts optional from / to to unfreeze a range of layers, or layer_names to unfreeze by name. If none are provided, all layers are unfrozen.

Usage

ggml_unfreeze_weights(
  model,
  from = 1L,
  to = length(model$layers),
  layer_names = NULL,
  ...
)

Value

The model with selected layers unfrozen.

Arguments

model

A model object (ggml_sequential_model or ggml_functional_model)

from

Integer index of the first layer to unfreeze (default: 1)

to

Integer index of the last layer to unfreeze (default: last layer)

layer_names

Character vector of layer names to unfreeze (overrides from/to)

...

Additional arguments passed to methods

Examples

Run this code
# \donttest{
model <- ggml_model_sequential() |>
  ggml_layer_dense(64, activation = "relu") |>
  ggml_layer_dense(10, activation = "softmax")

model <- ggml_freeze_weights(model)
model <- ggml_unfreeze_weights(model, from = 2)  # unfreeze last layer only
# }

Run the code above in your browser using DataLab