search

Susan19900316 / yolov5_tensorrt_int8

yolov5 tensorrt int8量化方法汇总
55 stars 13 forks source link

QAT量化 #9

Open SuGuilin opened 7 months ago

SuGuilin commented 7 months ago

请问QAT量化部分的代码和train.py一样，是如何进行量化的呀

Susan19900316 commented 7 months ago

image关注一下qat训练代码中加入的这些量，在训练中会直接用int8相关算子替换原始的算子，导出的onnx可以看到