ttslearn.train_util.num_trainable_params¶

ttslearn.train_util.num_trainable_params(model)[source]¶

Count the number of trainable parameters in the model.

Parameters: model (torch.nn.Module) – Model to count the number of trainable parameters.
Returns: Number of trainable parameters.
Return type: int