search

TexasInstruments / edgeai-mmdetection

This repository has been moved. The new location is in https://github.com/TexasInstruments/edgeai-tensorlab
https://github.com/TexasInstruments/edgeai
Other
0 stars 0 forks source link

QAT model configuration #9

Open ginamathew opened 2 years ago

ginamathew commented 2 years ago

Hi,

I would like to do model inference and pytorch to onnx conversion of custom object detection model(not in mmdetection) after QAT. Can you please help me in sharing sample code for the same.

I have model.py before QAT and as I understand this configuration get changed after doing QAT. I would like to know on how to change the model.py after QAT, for inference and also for onnx conversion.

regards, Gina

mathmanu commented 2 years ago

Please refer to the QAT documentation here: https://github.com/TexasInstruments/edgeai-torchvision/blob/master/docs/pixel2pixel/Quantization.md

You can also check how QAT is performed in these scripts: https://github.com/TexasInstruments/edgeai-torchvision/blob/master/references/edgeailite/scripts/quantize_classification_example.py https://github.com/TexasInstruments/edgeai-torchvision/blob/master/references/edgeailite/engine/train_classification.py