| countParams | R Documentation |
Count the number of trainable parameters in a model.
countParams(model, test_data)
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. |
Count the number of trainable parameters in an instantiated model subclassed from nn_module().
Counts of model parameters.
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.