tensorflow / tensorflow/model-optimization

How to extract original model's weight from trained QAT model?

Open
#977 9 comments 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

feature request
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.8
Are you willing to contribute it (Yes/No):
Motivation

How to extract the original model's weight from the trained QAT model?
I want to extract the original model's weight and QAT params, so I can set int8 params in TensorRT manually.

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.

Research direction

The issue names TensorFlow 2.8, a trained QAT model, original weights, QAT parameters, and manual TensorRT int8 parameters, but no files or tests. Start by locating the QAT model and conversion documentation; done should be a confirmed extraction procedure or a clearly documented limitation.

Written by the indexing model from the issue text.

Assessment

Tech stack
python, tensorflow
Domain
machine-learning
Issue type
Feature
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
25/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.