首页 > 其他 > 详细

目标检测faster rcnn error == cudaSuccess (2 vs. 0) out of memory

时间:2018-06-16 00:38:54      阅读:625      评论:0      收藏:0      [点我收藏+]

想尝试 更深更强的网络,或者自己写了一个费显存的层,发现1080 ti的11G显存不够用了,老师报显存不够怎么办?

Check failed: error == cudaSuccess (2 vs. 0)  out of memory

调小batch_size的大小。

 

怎么调呢?

调这个文件:

experiments/cfgs/faster_rcnn_end2end.yml

 

EXP_DIR: faster_rcnn_end2end
TRAIN:
HAS_RPN: True
IMS_PER_BATCH: 1
BBOX_NORMALIZE_TARGETS_PRECOMPUTED: True
RPN_POSITIVE_OVERLAP: 0.7
#RPN_BATCHSIZE: 256
RPN_BATCHSIZE: 160
PROPOSAL_METHOD: gt
BG_THRESH_LO: 0.0
#BATCH_SIZE: 64
BATCH_SIZE: 32
#RPN_POST_NMS_TOP_N: 700
RPN_POST_NMS_TOP_N: 500
USE_OHEM: True
ASPECT_GROUPING: False
TEST:
HAS_RPN: True

 

 

当然,写代码的方式也可以。比如说写一个batch_norm层,来归一化参数。让每一层的参数没那么凌乱。不过这个减少的显存还是比较小的。

目标检测faster rcnn error == cudaSuccess (2 vs. 0) out of memory

原文:https://www.cnblogs.com/whu-zeng/p/9189368.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!