Closed zengjianyou closed 5 years ago
不知道我说的对不对,但是我在用Mask_Rcnn的时候也曾经遇到过这样的问题。 出现这样的问题,我遇到的有两种情况: 1.你的batch太小,把batch增大(比较困难,毕竟GPU内存是固定的) 2.你的warm_up learning rate(0.001)太高了。把warm_up learning rate 调低,或者把warm up模式改成线性增长,把阀值设置为1.1(作者这个功能做没做我没仔细看,做了的话比较推荐线性增长做法)
@zengjianyou @NogizakaDaisuki 我自己把代码重新下载了一遍,然后跑上述例子没有出现问题,提出这个issue的已经联系过我了,我把自己生成的lmdb发给了他,让他试试看是否还有问题。
大神您好, 我用VGG16_voc2007_512.py训练voc0712数据,出现arm_loss=nan的问题,我截屏给您看看,