src.ml.utils.count_model_params¶

count_model_params(model)[source]¶

Count the number of parameters in a model.

Parameters: model – Pytorch model.
Returns: Number of parameters in the model.
Return type: int
Return type: int