摘要:在 cifar-10 上训练 AlexNet。将 32×32 的图片 resize 到 224×224 之后,数据将无法完全加载到 32G 内存中,会产生 OOM。那么此时的做法有: 1)将 resize 作为模型的一部分,如设置一个 layer 来对一个 batch 的图像进行 resize,这样 32×32 的 cifar-10 仍然可以完全加载到内存中; 2)一种通用的方法,每次只加载一部分数据到内存中,其余数据等到需要的时候再加载到内存。 阅读全文
posted @ 2019-07-15 22:18 wuliytTaotao 阅读 (256) 评论 (0) 编辑