search

yijingru / BBAVectors-Oriented-Object-Detection

[WACV2021] Oriented Object Detection in Aerial Images with Box Boundary-Aware Vectors
MIT License
462 stars 87 forks source link

作者，你好，关于hm loss 训练25个epoch 以后会出现nan的情况 #94

Open njauwhr opened 3 years ago

njauwhr commented 3 years ago

只用了一个titan, batch=10 学习率是1.25e-4，想知道hm为nan应该怎么解决，我把数据裁成1024的大小，是数据的问题吗？

yijingru commented 3 years ago

batch size过小会引发这个问题，猜测可能是batch图片分布非常不相似，类别不平衡，有一些图片甚至都没有物体。一般是训练后期才出现nan，猜测可能与adam有关

gzyluck commented 1 year ago

老哥，你咋解决的啊

gzyluck commented 1 year ago

只用了一个titan, batch=10 学习率是1.25e-4，想知道hm为nan应该怎么解决，我把数据裁成1024的大小，是数据的问题吗？

老哥，你咋解决的啊。我已经去掉没有目标的图片了。还是不行

kobezy commented 7 months ago

作者你好，请问训练随机轮次出现hm loss=0这个问题具体是如何解决的？非常期待您的答复。