model-optimization
model-optimization copied to clipboard
A toolkit to optimize ML models for deployment for Keras and TensorFlow, including quantization and pruning.
Remove optimizer to compare model size and compare the zipped model size to pass the test. And enable test because now it passed.
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...
Updates security files
**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...
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...
**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...
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...
**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...