yolov3 训练coco数据(cuda8.0-gxt1080)

1. train遇到这个问题:
已放弃Resizing
608
CUDA Error: out of memory
darknet: ./src/cuda.c:36: check_error: Assertion `0' failed.

百度的说法是gpu架构和作者默认使用的不一致。所以根据自己的gpu型号:gxt1080(pascal支持,百度即可得知...)。然后上官网可以查到gpu对应的comput_*:


但是呢我把这个修改好了,再train的时候,还是出现resizing * 出现 out of memory.....


难道是gpu的显存不够?....   或许吧.....

所以解决的方法就是:根据yolo训练自己的data的解答:

1:CUDA Error: out of memory表明显存不够,调小batch数值

2:Resizing到不同尺度,同样增加显存。设置random=0。关闭多尺度训练。



评论 15
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值