小編給大家分享一下Tensorflow如何實現(xiàn)釋放內(nèi)存,相信大部分人都還不怎么了解,因此分享這篇文章給大家參考一下,希望大家閱讀完這篇文章后大有收獲,下面讓我們一起去了解一下吧!
問題:
with tf.Session() as sess: saver = tf.train.Saver() #保存運算圖
當(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/'))
的時候由于session和graph沒有釋放內(nèi)存的運算圖就產(chǎn)生了報錯,
解決辦法:
在with tf.Session() as sess: 之后同時也要在with的范圍以外(注意),添加
tf.reset_default_graph()
代碼來重置默認的圖,這樣就能解決我下一步執(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)生的問題了。
以上是“Tensorflow如何實現(xiàn)釋放內(nèi)存”這篇文章的所有內(nèi)容,感謝各位的閱讀!相信大家都有了一定的了解,希望分享的內(nèi)容對大家有所幫助,如果還想學(xué)習(xí)更多知識,歡迎關(guān)注創(chuàng)新互聯(lián)行業(yè)資訊頻道!
網(wǎng)站欄目:Tensorflow如何實現(xiàn)釋放內(nèi)存-創(chuàng)新互聯(lián)
文章分享:http://chinadenli.net/article12/pppgc.html
成都網(wǎng)站建設(shè)公司_創(chuàng)新互聯(lián),為您提供外貿(mào)建站、網(wǎng)站改版、域名注冊、定制網(wǎng)站、品牌網(wǎng)站制作、企業(yè)建站
聲明:本網(wǎng)站發(fā)布的內(nèi)容(圖片、視頻和文字)以用戶投稿、用戶轉(zhuǎn)載內(nèi)容為主,如果涉及侵權(quán)請盡快告知,我們將會在第一時間刪除。文章觀點不代表本網(wǎng)站立場,如需處理請聯(lián)系客服。電話:028-86922220;郵箱:631063699@qq.com。內(nèi)容未經(jīng)允許不得轉(zhuǎn)載,或轉(zhuǎn)載時需注明來源: 創(chuàng)新互聯(lián)
猜你還喜歡下面的內(nèi)容