model-optimization

model-optimization copied to clipboard

Allow quantization of tied weights

2

**System information** - TensorFlow version (you are using): 2.6.0 (TFMOT 0.7.2) - Are you willing to contribute it (Yes/No): Potentially, with some advice on how to implement it **Motivation** Numerous...

hunse

Quantization aware training for a Transfer Learning MobileNet model

5

Hello, I have a MobileNetV2 That I am trying to use for image classification by means of transfer learning, although apparently seems to not work. Initially, I perform transfer learning...

publioelon

How to extract original model's weight from trained QAT model?

9

System information TensorFlow version (you are using): 2.8 Are you willing to contribute it (Yes/No): Motivation How to extract the original model's weight from the trained QAT model? I want...

doomooo

Make the utils module public available.

copybara-service[bot]

PCQAT not working if Conv2D has kernel size 1x1

8

**Describe the bug** When doing the [Sparsity and cluster preserving quantization aware training (PCQAT) Keras example](https://www.tensorflow.org/model_optimization/guide/combine/pcqat_example), if I use a Conv2D Layer with a kernel size of `(1, 1)` the...

YannPourcenoux

Mismatch in number of weights when loading quantized model (activation layer)

9

**Describe the bug** Saving and subsequently loading a quantized model results in the following error: ``` Traceback (most recent call last): File "test.py", line 37, in model = tf.keras.models.load_model('MinimalExample.h5') File...

Lotte1990

Structural (filter) pruning for convolutional layers

5

**System information** - TensorFlow version (you are using): 2.5.0 - Are you willing to contribute it (Yes/No): Yes **Motivation** Deciding on where to have high filter/channel counts in convnets can...

marcelroed

block_size for pruning 3D weights

1

**System information** Ubuntu 18.02 - TensorFlow version (you are using): 2.4.1 - Are you willing to contribute it (Yes/No): Yes This relates to the pruning optimization "block_size: The dimensions (height,...

eejlny

QAT (quantization aware training) Support quantizing models recursively

24

**Describe the bug** I'm doing transfer learning and would like to (at the end) quantize my model. The problem is that when I try to use the _quantize_model()_ function (which...

CRosero

Updated RELEASE notes with ARM contributions for TFMOT.

wwwind