Get model info and number of parameters
Information on the model can be accessed by the .summary()
method
from keras.models import load_model
model = load_model("my_model.h5")
model.summary()
_________________________________________________________________
Layer (type) Output Shape Param #
=================================================================
flatten_2 (Flatten) (None, 784) 0
_________________________________________________________________
dense_3 (Dense) (None, 128) 100480
_________________________________________________________________
dense_4 (Dense) (None, 10) 1290
=================================================================
Total params: 101,770
Trainable params: 101,770
Non-trainable params: 0
_________________________________________________________________