wuliytTaotao

摘要： 在 cifar-10 上训练 AlexNet。将 32×32 的图片 resize 到 224×224 之后，数据将无法完全加载到 32G 内存中，会产生 OOM。那么此时的做法有： 1）将 resize 作为模型的一部分，如设置一个 layer 来对一个 batch 的图像进行 resize，这样 32×32 的 cifar-10 仍然可以完全加载到内存中； 2）一种通用的方法，每次只加载一部分数据到内存中，其余数据等到需要的时候再加载到内存。 阅读全文