countParams: countParams

View source: R/utilities.R

countParamsR Documentation

countParams

Description

Count the number of trainable parameters in a model.

Usage

countParams(model, test_data)

Arguments

model

instantiated model object as nn_module subclass as opposed to luz fitted object.

test_data

Example tensor of the correct shape for the model being explored.

Details

Count the number of trainable parameters in an instantiated model subclassed from nn_module().

Value

Counts of model parameters.


geodl documentation built on Dec. 2, 2025, 5:06 p.m.