ValueError: Unknown layer: PruneLowMagnitude. Please ensure this object is passed to the `custom_objects` argument. See https://www.tensorflow.org/guide/keras/save_and_serialize#registering_the_custom_object for details.

majianjia / nnom

A higher-level Neural Network library for microcontrollers.

Apache License 2.0

909 stars 245 forks source link

ValueError: Unknown layer: PruneLowMagnitude. Please ensure this object is passed to the `custom_objects` argument. See https://www.tensorflow.org/guide/keras/save_and_serialize#registering_the_custom_object for details. #158

Open rxin-1219 opened 2 years ago

rxin-1219 commented 2 years ago

Hello, I would like to do Model optimization to trim non-essential weights at model , this not supported by the pruning API provided by tensorflow , is there a trimming function available?

majianjia commented 2 years ago

Hi @rxin-1219 , I don't think we support that, it will need to modify the backend ops which doesn't allow in cmsis-nn.