search

sfzhang15 / RefineDet

Single-Shot Refinement Neural Network for Object Detection, CVPR, 2018
Other
1.43k stars 393 forks source link

我用voc0712数据训练出现问题 #156

Closed zengjianyou closed 5 years ago

zengjianyou commented 5 years ago

大神您好， 我用VGG16_voc2007_512.py训练voc0712数据，出现arm_loss=nan的问题，我截屏给您看看，

1 然后在用自己的数据跑了下，refinedet-caffe也出现arm_loss类似问题，学习率也调了，但是arm_loss=nan

NogizakaDaisuki commented 5 years ago

不知道我说的对不对，但是我在用Mask_Rcnn的时候也曾经遇到过这样的问题。 出现这样的问题，我遇到的有两种情况： 1.你的batch太小，把batch增大（比较困难，毕竟GPU内存是固定的） 2.你的warm_up learning rate(0.001)太高了。把warm_up learning rate 调低，或者把warm up模式改成线性增长，把阀值设置为1.1（作者这个功能做没做我没仔细看，做了的话比较推荐线性增长做法）

sfzhang15 commented 5 years ago

@zengjianyou @NogizakaDaisuki 我自己把代码重新下载了一遍，然后跑上述例子没有出现问题，提出这个issue的已经联系过我了，我把自己生成的lmdb发给了他，让他试试看是否还有问题。