in Education by (55.8k points)

How to find Number of parameters of a keras model?

Is there a simple way to find the number of parameters of a keras model if I have CNN, LSTm etc. Just like we do it in FFN.

1 Answer

0 votes
by (55.8k points)
To find number of parameters of a Keras model just use:

model.count_params()

Or perform this directly:

import keras.backend as K

def size(model): # Compute number of params in a model (the actual number of floats)

    return sum([np.prod(K.get_value(w).shape) for w in model.trainable_weights])

Related questions

0 votes
asked Jan 31 in Education by JackTerrance (1.8m points)
...