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，即圖像的尺寸太大，佔用太大內存，建議調小尺寸。