Keras 快速解決OOM超內(nèi)存的問(wèn)題
如果在Keras內(nèi)部多次使用同一個(gè)Model，例如在不同的數(shù)據(jù)集上訓(xùn)練同一個(gè)模型進(jìn)而得到結(jié)果，會(huì)存在內(nèi)存泄露的問(wèn)題。在運(yùn)行幾次循環(huán)之后，就會(huì)報(bào)錯(cuò)OOM。
解決方法是在每個(gè)代碼后面接clear_session()函數(shù)，顯示的關(guān)閉TFGraph，再重啟。
詳情參考 https://keras.io/backend/#backend-functions。
from keras import backend as K
K.clear_session()
補(bǔ)充知識(shí)：Keras多次加載模型進(jìn)行測(cè)試時(shí)內(nèi)存溢出的解決方法
在進(jìn)行實(shí)驗(yàn)的過(guò)程中，保存了每個(gè)epoch的模型參數(shù)文件，需要驗(yàn)證每個(gè)保存的模型的效果，想到直接簡(jiǎn)單粗暴的手法，就是一個(gè)循環(huán)里加載模型并進(jìn)行驗(yàn)證，但是導(dǎo)致隨著加載的模型越來(lái)越多，速度越來(lái)越慢。
方法如下：在每次加載模型之前，清空模型占用的內(nèi)存即可。
import tensorflow as tf from keras import backend as K K.clear_session() tf.reset_default_graph() ‘'‘加載模型位置'‘'
更多內(nèi)容信息，可以參見(jiàn)官網(wǎng)介紹https://keras.io/api/utils/backend_utils/
以上這篇Keras 快速解決OOM超內(nèi)存的問(wèn)題就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
相關(guān)文章
django admin 自定義替換change頁(yè)面模板的方法
今天小編就為大家分享一篇django admin 自定義替換change頁(yè)面模板的方法，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2019-08-08
Python實(shí)現(xiàn)操縱控制windows注冊(cè)表的方法分析
這篇文章主要介紹了Python實(shí)現(xiàn)操縱控制windows注冊(cè)表的方法,結(jié)合實(shí)例形式分析了Python使用_winreg模塊以及win32api模塊針對(duì)Windows注冊(cè)表操作相關(guān)實(shí)現(xiàn)技巧,需要的朋友可以參考下2019-05-05
在深度學(xué)習(xí)中,語(yǔ)音的輸入都是需要處理的,下面這篇文章主要給大家介紹了關(guān)于python語(yǔ)音信號(hào)處理的相關(guān)資料,文中通過(guò)實(shí)例代碼介紹的非常詳細(xì),需要的朋友可以參考下2023-01-01
web自動(dòng)化測(cè)試Selenium點(diǎn)擊元素的常用方法
在Web自動(dòng)化測(cè)試中,Selenium提供多種點(diǎn)擊方法,常用的click()方法通過(guò)選中元素并觸發(fā)點(diǎn)擊事件,若click()方法不穩(wěn)定,可以采用JavaScript執(zhí)行點(diǎn)擊或使用ActionChains類模擬鼠標(biāo)點(diǎn)擊,需要的朋友可以參考下2024-09-09
python實(shí)現(xiàn)給scatter設(shè)置顏色漸變條colorbar的方法
今天小編就為大家分享一篇python實(shí)現(xiàn)給scatter設(shè)置顏色漸變條colorbar的方法，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2018-12-12
Python Flask的request對(duì)象使用詳解
本文介紹Flask request對(duì)象,一個(gè)完整的HTTP請(qǐng)求,包括客戶端向服務(wù)端發(fā)送的Request請(qǐng)求和服務(wù)器端發(fā)送Response響應(yīng).為了能方便訪問(wèn)獲取請(qǐng)求及響應(yīng)報(bào)文信息,Flask框架提供了一些內(nèi)建對(duì)象,下面就來(lái)說(shuō)一下Flask針對(duì)請(qǐng)求提供內(nèi)建對(duì)象reques,需要的朋友可以參考一下2023-02-02
Python自動(dòng)化之實(shí)現(xiàn)桌面壁紙下載器
隨著計(jì)算機(jī)性能的提升，人們對(duì)計(jì)算機(jī)個(gè)性化的要求也越來(lái)越高了，自己使用的計(jì)算機(jī)當(dāng)然要設(shè)置成自己喜歡的風(fēng)格！本文就來(lái)用Python做個(gè)桌面壁紙下載器，需要的可以參考一下2022-11-11
淺談Python xlwings 讀取Excel文件的正確姿勢(shì)
這篇文章主要介紹了淺談Python xlwings 讀取Excel文件的正確姿勢(shì)，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2021-02-02
PyTorch實(shí)現(xiàn)重寫/改寫Dataset并載入Dataloader
這篇文章主要介紹了PyTorch實(shí)現(xiàn)重寫/改寫Dataset并載入Dataloader，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2020-07-07
讓Python程序定時(shí)執(zhí)行的8種方法整理
在日常工作中，我們常常會(huì)用到需要周期性執(zhí)行的任務(wù)，一種方式是采用?Linux?系統(tǒng)自帶的?crond?結(jié)合命令行實(shí)現(xiàn)，另外一種方式是直接使用Python。本文整理了一下?Python?定時(shí)任務(wù)的實(shí)現(xiàn)方式，希望對(duì)大家有所幫助2023-01-01