tensorflow / tensorflow/models

Quantization Aware Training - TF 1.15 using TF-Slim

Open
#8,816 2 comments 0 reactions 2 assignees View on GitHub

@marksandler2 is already working on this.

Since Jul 9, 2020.

models:research type:bug
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

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.