ResourceExhaustedError (see above for traceback): OOM when allocating tensor with shape[682112,156]

運行出錯結果如下: 出錯原因主要和自己的算法有關,在圖像識別的程序中,我的filters的數量分配如下: Filters的數量越多,同一張圖片經過這些filters提取特徵後佔用顯卡內存數量越大,經過這兩層衆多數量的filter卷積後,佔用的內存極大。另外: 圖片經過這些filters提取特徵後緩存在顯卡內存中,經過flaten展平,在經過一層包含1024個神經元的神經網絡學習,每一批(batch
相關文章
相關標籤/搜索