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

這是tensorflow 一個經常性錯誤,錯誤的原因在於:顯卡內存不夠。 解決方法就是降低顯卡的使用內存,途徑有以下幾種措施: 減少Batch 的大小 分析錯誤的位置,在哪一層出現顯卡不夠,比如在全連接層出現的,則降低全連接層的維度,把 改成 啥的 增加pool 層,降低整個網絡的維度。 修改輸入圖片的大小 總結以下:想進一切辦法降維,降低網絡的維度。 ...

2018-06-21 20:34 0 4872 推薦指數:

查看詳情

錯誤tensorflow.python.framework.errors_impl.OutOfRangeError的解決方案

近日,在使用CascadeRCNN完成目標檢測任務時,我在使用這個模型訓練自己的數據集時出現了如下錯誤: 具體如下截圖所示: 我使用的教程是這個鏈接:cascade r-cnn訓練和測試(tensorflow框架) 在出現這個錯誤的時候,以為是數據集的錯誤,在經過多次檢查 ...

Fri Jan 03 04:26:00 CST 2020 0 15534
 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM