Skip to content

Instantly share code, notes, and snippets.

@AkiyonKS
Created August 9, 2022 04:56
Show Gist options
  • Save AkiyonKS/fd1b1a3047e2f851931ea2b2d0bb68c8 to your computer and use it in GitHub Desktop.
Save AkiyonKS/fd1b1a3047e2f851931ea2b2d0bb68c8 to your computer and use it in GitHub Desktop.
Model: "model"
_________________________________________________________________
Layer (type) Output Shape Param #
=================================================================
input_1 (InputLayer) [(None, 300, 300, 3)] 0
prune_low_magnitude_block1_ (None, 300, 300, 64) 3522
conv1 (PruneLowMagnitude)
prune_low_magnitude_block1_ (None, 300, 300, 64) 73794
conv2 (PruneLowMagnitude)
prune_low_magnitude_block1_ (None, 150, 150, 64) 1
pool (PruneLowMagnitude)
prune_low_magnitude_block2_ (None, 150, 150, 128) 147586
conv1 (PruneLowMagnitude)
prune_low_magnitude_block2_ (None, 150, 150, 128) 295042
conv2 (PruneLowMagnitude)
prune_low_magnitude_block2_ (None, 75, 75, 128) 1
pool (PruneLowMagnitude)
prune_low_magnitude_block3_ (None, 75, 75, 256) 590082
conv1 (PruneLowMagnitude)
prune_low_magnitude_block3_ (None, 75, 75, 256) 1179906
conv2 (PruneLowMagnitude)
prune_low_magnitude_block3_ (None, 75, 75, 256) 1179906
conv3 (PruneLowMagnitude)
prune_low_magnitude_block3_ (None, 37, 37, 256) 1
pool (PruneLowMagnitude)
prune_low_magnitude_block4_ (None, 37, 37, 512) 2359810
conv1 (PruneLowMagnitude)
prune_low_magnitude_block4_ (None, 37, 37, 512) 4719106
conv2 (PruneLowMagnitude)
prune_low_magnitude_block4_ (None, 37, 37, 512) 4719106
conv3 (PruneLowMagnitude)
prune_low_magnitude_block4_ (None, 18, 18, 512) 1
pool (PruneLowMagnitude)
prune_low_magnitude_block5_ (None, 18, 18, 512) 4719106
conv1 (PruneLowMagnitude)
prune_low_magnitude_block5_ (None, 18, 18, 512) 4719106
conv2 (PruneLowMagnitude)
prune_low_magnitude_block5_ (None, 18, 18, 512) 4719106
conv3 (PruneLowMagnitude)
prune_low_magnitude_block5_ (None, 9, 9, 512) 1
pool (PruneLowMagnitude)
sequential (Sequential) (None, 10) 84956176
=================================================================
Total params: 114,381,359
Trainable params: 42,478,602
Non-trainable params: 71,902,757
_________________________________________________________________
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment