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

As suggested [here](https://www.tensorflow.org/model_optimization/guide/pruning), model pruning currently only provides benefits in model compression/size reduction. Further framework support is necessary to provide latency improvements in TF/TFLite.

feature request
technique:pruning

An instance of the centroids initializer [expects](https://github.com/tensorflow/model-optimization/blob/v0.7.1/tensorflow_model_optimization/python/core/clustering/keras/clustering_centroids.py#L47) to be initialized with `data_format`. However, instead of `data_format` it gets `num_channels` here: https://github.com/tensorflow/model-optimization/blob/v0.7.1/tensorflow_model_optimization/python/core/clustering/keras/cluster_wrapper.py#L218 I suppose that line is erroneous. This may cause...

bug

Prior to filing: check that this should be a bug instead of a feature request. Everything supported, including the compatible versions of TensorFlow, is listed in the overview page of...

bug

Basic implementation for ternarization

technique:qat

Prior to filing: check that this should be a bug instead of a feature request. Everything supported, including the compatible versions of TensorFlow, is listed in the overview page of...

bug

Duplicate 'for' word in code block comment.

technique:qat

Support for the LSTMPeephole package has been removed in response to its removal from keras. However, the tensorflow-optimization package has not been updated in Pypi. One of my models relies...

bug

### 1. System information - OS Platform and Distribution (e.g., Linux Ubuntu 16.04): Ubuntu 20.04 - TensorFlow installation (pip package or built from source): pip - TensorFlow library (version, if...

Internal change

technique:qat

**Describe the bug** Unable to load the saved model after applying quantization aware training. **System information** TensorFlow version (installed from source or binary): 2.2 TensorFlow Model Optimization version (installed from...

bug