model-optimization icon indicating copy to clipboard operation
model-optimization copied to clipboard
verified publisher icon tensorflow

Is it possible to implement post-training pruning?

Open PhoeniXuzoo opened this issue 4 years ago • 4 comments

MacOS

  • TensorFlow version (you are using): 2.3
  • Are you willing to contribute it (Yes/No): No

Motivation

The document just shows how to prune the model during training. Each time, we need to call model.fit. During the process, other weights will be modified. We just want to set some weights to zero. In the meantime, don't change other weights.

The feature have been implemented in Pytorch.

PhoeniXuzoo avatar Jan 20 '21 22:01 PhoeniXuzoo

Currently we are working on this @fredrec has more details

daverim avatar Mar 16 '21 07:03 daverim

So pruning doesn't work if we are not able to retrain the model after adding the prune_low_magnitude layer ?

Gojo1729 avatar Sep 17 '21 05:09 Gojo1729

Hello guys, any news on the topic ?

Assia17 avatar May 20 '22 14:05 Assia17