TensorFlow和Keras解決數據量過大內存溢出

將上萬張圖片的路徑一次性讀到內存中，本身實現一個分批讀取函數，在該函數中根據本身的內存狀況設置讀取圖片，只把這一批圖片讀入內存中，而後交給模型，模型再對這一批圖片進行分批訓練，由於內存通常大於等於顯存，因此內存的批次大小和顯存的批次大小一般不相同。python Tensorlow 在input.py裏寫get_batch函數。web def get_batch(X_train, y_train,

>>阅读原文<<