Layers not pruned after using prune_low_magnitude

General Discussion
,
#1

I’m using a Resnet50 model

image
image670×742 22.3 KB

Even after the pruning step, the weights doesn’t become 0’s.

Can anyone help me?

#3

Could you please refer to this thread , pruning guide and let us know if it helps? Thank you!

#5

Hi MOUNIKA, Sorry for late reply

You can reduce the pruning frequency to < 80 (40 steps * 2 epochs) to make pruning happen within the training steps. Simply set
…PolynomialDecay( … , frequency = 10)

The default frequency is 100, so without setting it explicitly, the pruning does not take place during your training process.