keras-faster-rcnn icon indicating copy to clipboard operation
keras-faster-rcnn copied to clipboard

Published 20 hours ago • verified publisher icon yizt

rcnn_class_loss训练到一半就会变成nan

Open huangxudong opened this issue 4 years ago • 7 comments

训练的自己制作的数据集，batch为1，lr为0.000，其他的基本没变

huangxudong avatar Dec 09 '20 12:12 huangxudong

微信图片_20201209201523

huangxudong avatar Dec 09 '20 12:12 huangxudong

微信图片_20201209201529

huangxudong avatar Dec 09 '20 12:12 huangxudong

望您有时间答复，谢谢！

huangxudong avatar Dec 09 '20 12:12 huangxudong

训练的自己制作的数据集，batch为1，lr为0.000，其他的基本没变

@huangxudong 您好，batch最好大于1， lr为0？

yizt avatar Dec 10 '20 00:12 yizt

训练的自己制作的数据集，batch为1，lr为0.000，其他的基本没变

@huangxudong 您好，batch最好大于1， lr为0？

batch设置成2显存就爆了，除了调整图像尺寸还可以调整哪些呀？lr为0.0003

huangxudong avatar Dec 10 '20 02:12 huangxudong

训练集损失到1.7就下不去了，验证集的损失不降反升，最后画不出来几个框，这可咋办呀博主

huangxudong avatar Dec 10 '20 09:12 huangxudong

训练的自己制作的数据集，batch为1，lr为0.000，其他的基本没变

@huangxudong 您好，batch最好大于1， lr为0？

batch设置成2显存就爆了，除了调整图像尺寸还可以调整哪些呀？lr为0.0003

可以将head 中的通道数都减少，使得batch size至少大于1

yizt avatar Dec 11 '20 02:12 yizt