ResourceExhaustedError (see above for traceback): OOM when allocating tensor with shape[682112,156]

运行出错结果如下: 出错原因主要和自己的算法有关,在图像识别的程序中,我的filters的数量分配如下: Filters的数量越多,同一张图片经过这些filters提取特征后占用显卡内存数量越大,经过这两层众多数量的filter卷积后,占用的内存极大。另外: 图片经过这些filters提取特征后缓存在显卡内存中,经过flaten展平,在经过一层包含1024个神经元的神经网络学习,每一批(batch
相关文章
相关标签/搜索