model-optimization icon indicating copy to clipboard operation
model-optimization copied to clipboard

A toolkit to optimize ML models for deployment for Keras and TensorFlow, including quantization and pruning.

Results 228 model-optimization issues
Sort by recently updated
recently updated
newest added

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...

bug

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...

bug

**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**...

bug

**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...

bug

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....

bug

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...

feature request

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...

feature request
technique:qat

**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...

bug

**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...

feature request

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...

bug