Ultra-Light-Fast-Generic-Face-Detector-1MB
Ultra-Light-Fast-Generic-Face-Detector-1MB copied to clipboard
训练过程中loss 出现为nan 与 inf 问题
利用这个网络训练其他数据集时loss函数出现了nan 和 inf 输出,导致无法正常训练;这个问题是loss函数造成的还是数据集问题呢,该如何解?
我也遇到了,可能是参数吧,我也是自己的数据集,在yolov3那边是ok 的
如果是自建的数据集可以试着先调整学习率
Refer to this Issue: https://github.com/Linzaer/Ultra-Light-Fast-Generic-Face-Detector-1MB/issues/182#issuecomment-915404168 Hope this helps