YonghaoHe / LFFD-A-Light-and-Fast-Face-Detector-for-Edge-Devices

A light and fast one class detection framework for edge devices. We provide face detector, head detector, pedestrian detector, vehicle detector......
MIT License
1.31k stars 330 forks source link

训练出现Nan #61

Open mifan0208 opened 4 years ago

mifan0208 commented 4 years ago

作者,你好,在训练过程中location_loss出现了Nan,请问你遇到过这样的问题嘛?正常吗?

YonghaoHe commented 4 years ago

@mifan0208 偶尔遇到过。 这里很有可能你的数据准备有些问题,不然就是学习率太大。是训练widerface么?

mifan0208 commented 4 years ago

对的,训练的widerface,找到原因了,是因为我计算loss的时候,先进行计算l2loss,再乘的mask;在训练的过程中,cls的metric不怎么提升,或者提升很慢,在0.05左右震荡(训了3个epoch),请问你当时的metric提升的怎么样呢?

mifan0208 commented 4 years ago

@YonghaoHe 你好,我在训练的时候cls_loss很低,但是metric却不高,很奇怪,能加一下你的联系方式嘛?咨询你个问题,谢谢。

121649982 commented 4 years ago

训练的脚本是哪 一个?