model-optimization
model-optimization copied to clipboard
A toolkit to optimize ML models for deployment for Keras and TensorFlow, including quantization and pruning.
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.
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...
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...
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...
Duplicate 'for' word in code block comment.
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...
### 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...
**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...