深度学习图像目标检测时。使用pytorch的detectron2平台调试faster-rcnn代码训练的时候,迭代到一段时间后就报错。FloatingPointError: Predicted boxes or scores contain Inf/Nan. Training has diverged.当训练bacthsize为1或者2时,报这个错误;把batchsize改大为4时,提示显存不足。
收起
把学习率改小些试试看
报告相同问题?
程序员都在用的中文IT技术交流社区
专业的中文 IT 技术社区,与千万技术人共成长
关注【CSDN】视频号,行业资讯、技术分享精彩不断,直播好礼送不停!