Tensorflow中迁移学习出现 OOM 解决方案

tensorflow.python.framework.errors_impl.ResourceExhaustedError: OOM when allocating tensor with shape[]

出现以上情况要注意以下两个方式：

1.batch_size值设置过大，导致内存溢出，batch_size是每次送入模型中的值，由于GPU的关系，一般设为16,32,64,128。

2. imagesize，即图像的尺寸太大，占用太大内存，建议调小尺寸。