在我的Dell Core i7-16GB RAM-4gb 960m GPU笔记本电脑上,我正在使用3d CNN对肺部CT图像进行分类的项目。我正在使用Tensorflow的CPU版本。图像准备为numpy数组大小(25,50,50)。
我的CNN模型具有2个转换层,2个maxpool层,1个FC层和输出层。通过这种架构,我可以训练大约(5000至6000)个样本的模型。添加更多层后,我的模型现在具有6个转换层,3个最大池层,FC和输出层。我的问题是在更改架构后,仅使用了1000多个样本,我的内存被填满,并且出现内存错误。我试图做较小的批次,但是每次都会出现相同的错误。我有两个问题:
为什么通过添加更多层,模型需要更多内存?
有什么办法可以解决这类问题?
这可能是因为您有很多参数,并且默认情况下您的GPU用于计算,而4GB可能不适合..
—
Aditya
我使用的是Tensorflow的CPU版本而不是GPU
—
Hunar A.Ahmed