nnsvs.train_util.num_trainable_params

nnsvs.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