n_trainable¶

nyx.core.paramtree.n_trainable(tree, *, include_frozen=False)[source]¶

Number of free scalar values.

Parameters:

tree (pytree)
include_frozen (bool) – Whether to count frozen values too.

Return type: