Quantization official example


Is there any official quantization training examples?

We have examples using the previous eager mode quantization: Quantization Recipe — PyTorch Tutorials 2.0.1+cu117 documentation and https://github.com/pytorch/vision/blob/main/references/classification/train_quantization.py

but @andrewor is working on a tutorial for the new quantization flow ((prototype) PyTorch 2.0 Export Post Training Static Quantization — PyTorch Tutorials 2.0.1+cu117 documentation) as well

1 Like

Hi @jerryzh168 , @andrewor

I am interested in the comments in the link:
In addition, PyTorch also supports quantization aware training, which models quantization errors in both the forward and backward passes using fake-quantization modules.

Do you have example code/doc for my reference?

yeah the doc I linked is the example for eager mode quantization flow.


Dear Jerry,
what I am looking for is the quantization aware training. is your link the same as what I asked?

Yes, that’s correct. did you see that?