小編給大家分享一下Tensorflow如何實(shí)現(xiàn)釋放內(nèi)存,相信大部分人都還不怎么了解,因此分享這篇文章給大家參考一下,希望大家閱讀完這篇文章后大有收獲,下面讓我們一起去了解一下吧!
成都創(chuàng)新互聯(lián)主營(yíng)崇信網(wǎng)站建設(shè)的網(wǎng)絡(luò)公司,主營(yíng)網(wǎng)站建設(shè)方案,app軟件開(kāi)發(fā),崇信h5小程序定制開(kāi)發(fā)搭建,崇信網(wǎng)站營(yíng)銷推廣歡迎崇信等地區(qū)企業(yè)咨詢問(wèn)題:
with tf.Session() as sess: saver = tf.train.Saver() #保存運(yùn)算圖
當(dāng)我退出with并且立馬執(zhí)行
self.sess = tf.Session() self.saver = tf.train.import_meta_graph('./Model/model.ckpt.meta') self.saver.restore(self.sess, tf.train.latest_checkpoint('./Model/'))
的時(shí)候由于session和graph沒(méi)有釋放內(nèi)存的運(yùn)算圖就產(chǎn)生了報(bào)錯(cuò),
解決辦法:
在with tf.Session() as sess: 之后同時(shí)也要在with的范圍以外(注意),添加
tf.reset_default_graph()
代碼來(lái)重置默認(rèn)的圖,這樣就能解決我下一步執(zhí)行代碼
self.sess = tf.Session() self.saver = tf.train.import_meta_graph('./Model/model.ckpt.meta') self.saver.restore(self.sess, tf.train.latest_checkpoint('./Model/'))
所產(chǎn)生的問(wèn)題了。
以上是“Tensorflow如何實(shí)現(xiàn)釋放內(nèi)存”這篇文章的所有內(nèi)容,感謝各位的閱讀!相信大家都有了一定的了解,希望分享的內(nèi)容對(duì)大家有所幫助,如果還想學(xué)習(xí)更多知識(shí),歡迎關(guān)注創(chuàng)新互聯(lián)行業(yè)資訊頻道!