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

Remove optimizer to compare model size and compare the zipped model size to pass the test. And enable test because now it passed.

comp:weight compression

Hi Everyone, I tried to post quantize fastspeech2 model of TensorFlowTTS ,but got the error message, I was wondering if anyone knows the detials of this error and how to...

bug

Internal change

technique:pruning

**Describe the bug** Can't use latest TFMOT 0.8.0 with the latest TF 2.15.0.post1 on a fresh install because of missing `tf_keras` dependency. This complicates installations of downstream packages that depend...

bug

Hi all, I was trying to quantize my model but something strange popped up. I am using TensorFlow v2.14 and tfmot v0.7.5 I have a sub-classed tf.Keras.Model. It contains some...

bug

**System information** - TensorFlow version (you are using): 2.15 - Are you willing to contribute it (Yes/No): Yes **Motivation** I am trying to train RNN model with quantization aware training...

feature request

Doing 16x8 quantization on RNN models fails. **Code to reproduce the issue** [gist to reproduce the issue on Google Colab](https://colab.research.google.com/drive/1tkXarTrLDA468U1AKXbs6yqdrPcfIsZj?usp=sharing) Code: ```python import numpy as np import tensorflow as tf...

bug

**Describe the bug** Using the `MovingAverageQuantizer` with parameter `per_axis` set to True results in this error: > tensorflow_model_optimization\python\core\quantization\keras\quant_ops.py", line 335, in _FakeQuantWithMinMaxVars * > assert len(min_var.get_shape()) == 1 It's caused...

bug