這是tensorflow 一個經常性錯誤,錯誤的原因在於:顯卡內存不夠。
解決方法就是降低顯卡的使用內存,途徑有以下幾種措施:
1 減少Batch 的大小
2 分析錯誤的位置,在哪一層出現顯卡不夠,比如在全連接層出現的,則降低全連接層的維度,把2048改成1042啥的
3 增加pool 層,降低整個網絡的維度。
4 修改輸入圖片的大小
總結以下:想進一切辦法降維,降低網絡的維度。
這是tensorflow 一個經常性錯誤,錯誤的原因在於:顯卡內存不夠。
解決方法就是降低顯卡的使用內存,途徑有以下幾種措施:
1 減少Batch 的大小
2 分析錯誤的位置,在哪一層出現顯卡不夠,比如在全連接層出現的,則降低全連接層的維度,把2048改成1042啥的
3 增加pool 層,降低整個網絡的維度。
4 修改輸入圖片的大小
總結以下:想進一切辦法降維,降低網絡的維度。
本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。