yolov5prune icon indicating copy to clipboard operation
yolov5prune copied to clipboard

运行python train_sparsity.py --st --sr 0.0001 --weights yolov5s.pt --adam --epochs 100 训练出现Nan

Open liufqing opened this issue 3 years ago • 8 comments

image

liufqing avatar Sep 07 '21 01:09 liufqing

我也遇到了同样的问题。 同求解

Yzh619 avatar Sep 17 '21 01:09 Yzh619

请问楼主解决了吗,我也遇到了这个问题

yezechen avatar Oct 21 '21 06:10 yezechen

请问楼主解决了吗,我也遇到了这个问题

你尝试一下把sr再缩小10倍再训练

Yzh619 avatar Oct 21 '21 07:10 Yzh619

请问楼主解决了吗,我也遇到了这个问题

你尝试一下把sr再缩小10倍再训练

我试一下,谢谢

yezechen avatar Oct 21 '21 07:10 yezechen

请问楼主解决了吗,我也遇到了这个问题

你尝试一下把sr再缩小10倍再训练

我试一下,谢谢

你好,请问这个问题解决了吗?我也遇到了相同的问题

Rainlolthx avatar Nov 16 '21 01:11 Rainlolthx

请问楼主解决了吗,我也遇到了这个问题

你尝试一下把sr再缩小10倍再训练

我试一下,谢谢

你好,请问这个问题解决了吗?我也遇到了相同的问题

调整SR的值了嘛,这个你得自己尝试调参。

Yzh619 avatar Nov 16 '21 01:11 Yzh619

请问楼主解决了吗,我也遇到了这个问题

你尝试一下把sr再缩小10倍再训练

我试一下,谢谢

你好,请问这个问题解决了吗?我也遇到了相同的问题

调整SR的值了嘛,这个你得自己尝试调参。

有在调整sr值,目前还没有结果,我是在epoch20-40之间会出现nan的情况,想明白可能是什么原因导致的

Rainlolthx avatar Nov 16 '21 03:11 Rainlolthx

有遇到这个问题能不能贴一张bn的直方图? 会不会是因为sr太大,导致某一层bn都逼近0,从而该层之后响应为0?

midasklr avatar Dec 14 '21 08:12 midasklr