PyTorch Quantization

quantization
1

Hi all,
I am trying the resnet50 model quantization with PyTorch and I tried these 3 lines of code :
the import, model=qn.resnet50(pretrain=true), and model.state_dict()), and why the coefficients being shown are all float values if this is the quarantined version of the model?
Noticed this while trying to figure out how to save/load the coefficients for a quantized model, and is there anything special you need to do convert model coefficients between float and int8.

Please let me know. Appreciate any help/suggestions , Thanks!

2

You need to set quantized=True when you load the model. i.e do model =qn.resnet(pretrained=True, quantized=True)