dmlcloud.count_parameters

dmlcloud.count_parameters(module)

Returns the number of trainable parameters in a module.

Parameters:

module (nn.Module) – The module to count the parameters of.

Returns:

The number of trainable parameters.

Return type:

int