tensorflow 運行時內存持續增長(溢出)

tensorflow 訓練部分代碼幾乎都採用循環,如果在循環中用到tf的op操作,就會導致節點越來越多,圖的大小也不停的增大,故而內存會持續增長 常見情況有幾種 1、sess.run(a+b)       寫成sess.run(add_a_b)最好 2、tf.train.Saver() 3.不要使用額外的圖 原因是在對每一張圖片正向傳播結束後,都會在留下中間信息。具體地說是在我將正向傳播的代碼與模
相關文章
相關標籤/搜索