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

tensorflow / model-optimization

A toolkit to optimize ML models for deployment for Keras and TensorFlow, including quantization and pruning.

https://www.tensorflow.org/model_optimization

Apache License 2.0

1.49k stars 319 forks source link

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

Closed zhenyu-ponyai closed 2 years ago

zhenyu-ponyai commented 2 years ago

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.

zhenyu-ponyai commented 2 years ago

Hi, I wonder how to extract the original model's weight from the trained QAT model? @thaink