YOLOv5 Series Multi-backbone(TPH-YOLOv5, Ghostnet, ShuffleNetv2, Mobilenetv3Small, EfficientNetLite, PP-LCNet, SwinTransformer YOLO), Module(CBAM, DCN), Pruning (EagleEye, Network Slimming), Quantization (MQBench) and Deployment (TensorRT, ncnn) Compression Tool Box.
989
stars
201
forks
source link
运行 YOLOv5s模型在NNIE量化训练下错误！ #91
Open
tensorflowt opened 1 year ago
您好！
我这边最近尝试复现您开源的基于MQBench下的NNIE量化方式，为了解决在海思芯片上算法量化精度掉点的问题。
然后，我这边复现的时候遇到了如下问题：
1.common.py中C3模块的问题，导致报错，修改为官方v6.0版本的C3模块解决。
2.AttributeError: 'GraphModule' object has no attribute 'module'
暂未解决，我看到其他开发者也遇到了类似问题，如下：
https://github.com/Gumpest/YOLOv5-Multibackbone-Compression/issues/83
能帮忙指导一下吗？谢谢~