ZJU-lishuang / yolov5_prune

yolov5 prune,Support V2, V3, V4 and V6 versions of yolov5
Apache License 2.0
559 stars 137 forks source link

剪枝后map显示nan #45

Open haoxue1215 opened 3 years ago

haoxue1215 commented 3 years ago

屏幕截图 2021-06-28 153924

您好,我使用 slim_prune_yolov5s.py对稀疏后的模型剪枝,得到剪枝后的cfg和pt文件了,但是map显示为nan,无论剪枝率设置多少都是这样,请问这边是出现什么问题了吗?

ZJU-lishuang commented 3 years ago

目前重新整理复现了改工程,可以重新尝试一下