search

THU-MIG / torch-model-compression

针对pytorch模型的自动化模型结构分析和修改工具集，包含自动分析模型结构的模型压缩算法库
MIT License
244 stars 40 forks source link

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

Open DWendou opened 2 years ago

lix19937 commented 1 year ago

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