ggmlR (version 0.6.1)

ggml_load_model: Load a Full Model (Architecture + Weights)

Description

Restores a model previously saved with ggml_save_model(). The returned model is compiled and ready for ggml_predict() / ggml_evaluate(). Call ggml_fit() again to continue training.

Usage

ggml_load_model(path, backend = "auto")

Value

A compiled model object.

Arguments

path

File path to an RDS file written by ggml_save_model().

backend

Backend selection: "auto", "cpu", or "vulkan".

Examples

# \donttest{
model <- ggml_model_sequential() |>
  ggml_layer_dense(16L, activation = "relu", input_shape = 4L) |>
  ggml_layer_dense(2L,  activation = "softmax")
model <- ggml_compile(model, optimizer = "adam",
                       loss = "categorical_crossentropy")
x <- matrix(runif(64 * 4), 64, 4)
y <- matrix(c(rep(c(1,0), 32), rep(c(0,1), 32)), 64, 2)
model <- ggml_fit(model, x, y, epochs = 1L, batch_size = 32L, verbose = 0L)
tmp <- tempfile(fileext = ".rds")
ggml_save_model(model, tmp)
model2 <- ggml_load_model(tmp)
# }