''tensorflow.python.framework.errors_impl.ResourceExhaustedError: OOM when allocating tensor with sh

這是tensorflow 一個常常性錯誤,錯誤的緣由在於:顯卡內存不夠。網絡

解決方法就是下降顯卡的使用內存,途徑有如下幾種措施:圖片

1 減小Batch 的大小內存

2 分析錯誤的位置,在哪一層出現顯卡不夠,好比在全鏈接層出現的,則下降全鏈接層的維度,把2048改爲1042啥的tensorflow

3 增長pool 層,下降整個網絡的維度。方法

4 修改輸入圖片的大小總結

 

總結如下:想進一切辦法降維,下降網絡的維度。錯誤

相關文章
相關標籤/搜索