我目前正在为视频分类建立3D卷积网络。主要问题是我太容易耗尽内存。即使将batch_size设置为1,也仍然没有足够的内存来按我的方式训练CNN。
我正在使用具有4Gb VRAM的GTX 970(3.2 Gb可供Tensorflow免费使用)。我希望它仍然可以训练我的网络,也许使用我的RAM存储器作为备份,或者分部分进行计算。但是直到现在,我只能运行它来简化CNN,这直接影响性能。
我认为我可以在CPU上运行,但是速度明显较慢,因此也不是一个好的解决方案。
有没有比购买更好的GPU更好的解决方案?
提前致谢。
使用梯度检查点将有助于限制内存。