Skip to content

Instantly share code, notes, and snippets.

@sayakpaul

sayakpaul/pruned_model_summary.py

Created June 21, 2020 15:50
Show Gist options
  • Save sayakpaul/5a467b8f994c238df72fc08c80e85b43 to your computer and use it in GitHub Desktop.
Save sayakpaul/5a467b8f994c238df72fc08c80e85b43 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
pruned_model_summary.py
Layer (type) Output Shape Param #
=================================================================
prune_low_magnitude_conv2d ( (None, 26, 26, 12) 230
_________________________________________________________________
prune_low_magnitude_max_pool (None, 13, 13, 12) 1
_________________________________________________________________
prune_low_magnitude_flatten (None, 2028) 1
_________________________________________________________________
prune_low_magnitude_dense (P (None, 10) 40572
=================================================================
Total params: 40,804
Trainable params: 20,410
Non-trainable params: 20,394
_________________________________________________________________
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment