Pruning error of transfer learning model MobilenetV3 Large !

[somum]

Trying to prune MobilenetV3 Large model after training but showing an error of -

ValueError: Please initialize Prune with a supported layer. Layers should either be supported by the PruneRegistry (built-in keras layers) or should be a PrunableLayer instance, or should has a customer defined get_prunable_weights method. You passed: <class 'tensorflow.python.keras.layers.preprocessing.image_preprocessing.Rescaling'>

I was following the guide of Tensorflow here.

[rino20]

Hi @somum

This has been resolved in https://github.com/tensorflow/model-optimization/pull/754 Could you sync your tfmot with HEAD or nightly and re-test the code?