Tensorflow 程序 內存泄漏+顯存分配不足 數據預處理

    1. 目的                   在循環中,從硬盤中讀取每個batch 的圖片。     2. 錯誤步驟及原因                   圖片截取從here。              3. 修改方法     方法一:可以將圖片數據預處理部分不使用 tensor ,用的python 常見的numpy 等類型,這樣就不會產生新的tensor 結點。     方法二:使
相關文章
相關標籤/搜索