NVIDIA-AI-IOT / yolo_deepstream

yolo model qat and deploy with deepstream&tensorrt
Apache License 2.0
533 stars 135 forks source link

Can you make a version of yolox QAT ? #32

Closed DWendou closed 1 year ago

wanghr323 commented 1 year ago

Errr, haha...... that is just a sample code to show how to use QAT with https://github.com/NVIDIA/TensorRT/tree/main/tools/pytorch-quantization You can have a try following our code.