tensorflow / tensorflow/models
Quantization Aware Training - TF 1.15 using TF-Slim
@marksandler2 is already working on this.
Since Jul 9, 2020.
- Dominant language
- Python
- Stars
- 77.7k
- Forks
- 44.8k
- PR merge metrics
- No merged PRs in 30d
Description
I am doing quantization aware training for Mobilenetv2 using TF Slim (TF - 1.15).
For the full model (without quantization) I get an accuracy of 80% with 1200 steps while the quantization aware trained model is still at 40% accuracy after 40,000 steps.
In order to activate quantization aware training, I set --quantize_delay=1. Do I need to do something else??
While looking for this issue, I found this : tensorflow/model-optimization#368. Will this issue affect tf-slim as well?
I am trying to use tf-slim to generate a quantization aware model with uint8 input as TF 2 only supports float input for quantization aware models.
If anyone has any other suggestions on quantization aware training with uint8 input of Mobilenetv2, please let me know!
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.
Assessment
This issue has not been assessed yet.