search

ZJU-lishuang / yolov5_prune

yolov5 prune，Support V2, V3, V4 and V6 versions of yolov5
Apache License 2.0
556 stars 136 forks source link

剪枝前后的模型map都显示为nan #70

Open MagicIsGone1 opened 2 years ago

MagicIsGone1 commented 2 years ago

您好，根据您提供的示例进行操作，基础训练和稀疏化训练时都很正常，但在剪枝的时候出现了问题。剪枝前和剪枝后的map值都显示为nan，同时伴随以下报错。 1 多次调整剪枝率之后发现map要么为0要么就为nan。请问一下问题出在哪里

ZJU-lishuang commented 2 years ago

没碰到过相关问题