model-optimization
model-optimization copied to clipboard
A toolkit to optimize ML models for deployment for Keras and TensorFlow, including quantization and pruning.
Hi, **Describe the bug** A clear and concise description of what the bug is. Getting below error while trying to cluster the convolution layers. Is Conv1DTranspose clustering is possible or...
There has recently been an article saying mobilenetv3 can be used for QAT [https://blog.tensorflow.org/2022/06/Adding-Quantization-aware-Training-and-Pruning-to-the-TensorFlow-Model-Garden.html](https://blog.tensorflow.org/2022/06/Adding-Quantization-aware-Training-and-Pruning-to-the-TensorFlow-Model-Garden.html) However when I run this code: ``` from tensorflow.keras.applications import MobileNetV3Small import tensorflow_model_optimization as tfmot model...
**Describe the bug** Prune Low Magnitude seems not to update the weights to 0 (I am using Constant Sparsity), when using a small dataset for training (1000 Images). **System information**...
**Describe the bug** When using `prune_low_magnitude()`, my model is not pruned if the batch size is low. **System information** TensorFlow version (installed from source or binary): 2.8.0 installed via pip...
Just a quick question. I want my final model to be full int8 instead of float32 for input and outputs. I want the training to be as accurate as possible....
Hi, This is more of a question than a feature request which I don't know where else to post. So I'm trying to perform quantization aware training to a model...
Hello, It seems as though Conv1D layer quantization is not yet supported? I get an error while trying to quantize a model containing Conv1D layers..is this going to be supported...
**Describe the bug** When we try to quantize a model containing a `TFOpLambda` layer, an AttributeError `'list' object has no attribute 'dtype'` would occur. **System information** TensorFlow version (installed from...
**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...
Hi, is there any work around method to use tfmot.sparsity.keras.strip_pruning on subclass models? I tried to use prune_low_magnitude on subclass by apply it to the layers in subclasses as a...