yolov5prune icon indicating copy to clipboard operation
yolov5prune copied to clipboard

Published 20 hours ago • verified publisher icon midasklr

稀疏训练报错：ValueError: autodetected range of [nan, nan] is not finite

Open baidingyuan opened this issue 3 years ago • 2 comments

参数设置：sr：0.000001 问题：第二轮的时候所有的值都变成nan了。 猜想：之前train训练的时候迭代次数少，得到的权重文件是300多M，可以正常稀疏训练。新train迭代了300次，得到的权重跟官方预训练权重一致，都是170多M，稀疏训练猜想：之前train训练的时候迭代次数少，得到的权重文件是300多M，可以正常稀疏训练。新train迭代了300次，得到的权重跟官方预训练权重一致，都是170多M，稀疏训练nan报错。

baidingyuan avatar Jan 19 '22 11:01 baidingyuan

请问你解决了吗，我也遇到这个问题了，但是我是在60多轮时出现的这问题

1781686573 avatar Oct 09 '23 09:10 1781686573

请问解决了吗？我也遇到了，epoch=100，在第8轮遇到该问题

Sally-lxy avatar Jan 02 '24 05:01 Sally-lxy