Quantization official example

quantization
1

Hi,

Is there any official quantization training examples?

2

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
3

Hi @jerryzh168 , @andrewor

I am interested in the comments in the link:
https://pytorch.org/docs/stable/quantization.html#
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?

4

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

5

@jerryzh168

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

6

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