cnn的训练数据太大,内存溢出怎么办

如题,基于keras,cnn的训练数据太大,内存溢出,怎么分批输入训练数据,然后再指定batch_size进行分批次训练,fit_generator好像可以分批输入训练数据,但是不能指定batch_size,只是将分批输入的数据看作一个batch。

batch_size弄小一些(会增加训练时间/降低优化器精度)
模型简单一些,多一些pooling层,使用比较小的倦积核。
图像预处理,降低分辨率
换更好的硬件,比如买那种二手的计算卡(所谓矿渣卡),16GB/24GB县存的,特别实惠。