Android应用源码之加载本地图片，绝对不会出现OOM技术实现完整源码下载
2022-01-12 14:11:09
1.36MB
问题：
在使用mask_rcnn预测自己的数据集时，会出现下面错误：
ResourceExhaustedError: OOM when allocating tensor with shape[1,512,1120,1120] and type float on /job:localhost/replica:0/task:0/device:GPU:0 by allocator GPU_0_bfc
[[{{node rpn_model/rpn_conv_shared/convolution}} = Conv2D[T=DT_FLOAT, data_format="NCHW", dilation
2021-09-29 10:22:50
42KB
1