''tensorflow.python.framework.errors_impl.ResourceExhaustedError: OOM when allocating tensor with shape[?]'' 錯誤分析


這是tensorflow 一個經常性錯誤,錯誤的原因在於:顯卡內存不夠。

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

1 減少Batch 的大小

2 分析錯誤的位置,在哪一層出現顯卡不夠,比如在全連接層出現的,則降低全連接層的維度,把2048改成1042啥的

3 增加pool 層,降低整個網絡的維度。

4 修改輸入圖片的大小

 

總結以下:想進一切辦法降維,降低網絡的維度。


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM