yizt/keras-faster-rcnn

rcnn_class_loss训练到一半就会变成nan

huangxudong opened this issue · 7 comments

huangxudong commented

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

huangxudong commented

望您有时间答复，谢谢！

yizt commented

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

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

huangxudong commented

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

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

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

huangxudong commented

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

yizt commented

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

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

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

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