Keras,在lambda conv2d层中使用的海量内存

时间:2019-07-09 16:41:52

标签: tensorflow keras conv-neural-network

我正在执行模式匹配问题,其中有一个100x100像素的模板和1000x1000像素的图像。该模板用于生成在lambda层内的conv2d中使用的100x100x16滤镜(该滤镜用于通过某些CNN层生成的输入图像的1000x1000x16版本)。所有这些都适用于较小尺寸的东西,但是使用此完整尺寸,我的内存不足。该错误表明它正试图在某个地方分配[1,1000000,160000]张量。为什么它要分配这么大的东西,我对此有什么办法?

0 个答案:

没有答案