THU-MIG / torch-model-compression

针对pytorch模型的自动化模型结构分析和修改工具集，包含自动分析模型结构的模型压缩算法库

有没有实验yolo系列QAT感知量化训练模型？

DWendou opened this issue 2 years ago · comments

DWendou commented 2 years ago

有没有实验yolo系列QAT感知量化训练模型？

lix19937 commented a year ago

see tensorrt tools/pytorch-quantization, you can use quant.nn.module replace torch.nn.module