首页 PaddleCV 帖子详情
飞桨轻量级目标检测挑战赛,评分模型时异常
收藏
快速回复
PaddleCV 问答目标检测 922 4
飞桨轻量级目标检测挑战赛,评分模型时异常
收藏
快速回复
PaddleCV 问答目标检测 922 4

用coco数据集、shufflenetv2-ssd,也就是轻量级竞赛里的内容,epoch_num =8 或10,训练出的模型,使用score.py评分时报内存异常,提交到百度后台大概半小时左右可以出成绩(之前正常模型十几分钟),异常截图如下。同样代码和数据epoch_num=1训练出的模型评分正常。
aistudio(paddle 1.5.0)和本地(1.6.1)都出。

QQ图片20200117152008
image

0
收藏
回复
全部评论(4)
时间顺序
aistudio_2222
#3 回复于2021-05

mark

0
回复
时间女神
#4 回复于2021-12

这是显存溢出了,减小batch吧

0
回复
DeepGeGe
#5 回复于2021-12

感觉你这种情况是内存溢出,而不是显存溢出。因为你用了coco数据集,这个数据集是非常大的,所以不要一次性将所有图片加载到内存,而是按需读取。

0
回复
李长安
#6 回复于2021-12

把bs改成1试试

0
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户