Learn R Programming

ggmlR (version 0.6.1)

nn_build_functional_graph: Build ggml computation graph for a functional model

Description

Build ggml computation graph for a functional model

Usage

nn_build_functional_graph(model, batch_size, training = FALSE)

Value

Named list with inputs, outputs, ctx_weights, ctx_compute, buffer, node_weights

Arguments

model

A ggml_functional_model

batch_size

Integer batch size

training

Logical; TRUE during fit (activates dropout scaling), FALSE during evaluate/predict (dropout becomes identity).