tensorflow / tensorflow/model-optimization
Support for tensorflow hub layers
Nobody has claimed this yet.
- Dominant language
- Python
- Stars
- 1.6k
- Forks
- 349
- Avg merge
- 3d 2h
- Merged PRs (30d)
- 1
Description
System information
- TensorFlow version (you are using): 2.7
- Are you willing to contribute it (Yes/No): Yes
Motivation
Pruning and clustering both do not currently support tensorflow hub models.
Ex: ValueError: Please initialize Cluster with a supported layer. Layers should either be a ClusterableLayer instance, or should be supported by the ClusteringRegistry. You passed: <class 'tensorflow_hub.keras_layer.KerasLayer'>
Most of the NLP models which can benefit from these techniques are hosted on tf hub, hence, the request.
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
Start by reproducing the reported ValueError with TensorFlow 2.7 and a TensorFlow Hub KerasLayer, then inspect how pruning and clustering determine supported layers. Done means TensorFlow Hub models can be used with both techniques without the reported unsupported-layer error.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- keras, python, tensorflow
- Domain
- machine-learning
- Issue type
- Feature
- Difficulty
- 4/5
- Estimated time
- 3-5 days
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 35/100