keras的load_model實(shí)現(xiàn)加載含有參數(shù)的自定義模型
網(wǎng)上的教程大多數(shù)是教大家如何加載自定義模型和函數(shù)，如下圖
這個(gè)SelfAttention層是在訓(xùn)練過程自己定義的一個(gè)class，但如果要加載這個(gè)自定義層，需要在load_model里添加custom_objects字典，這個(gè)自定義的類，不要用import ，最好是直接復(fù)制進(jìn)再訓(xùn)練的模型中，這些是基本教程。
------------------分割線講重點(diǎn)------------------
如果直接運(yùn)行上面的代碼，會出現(xiàn)一個(gè)init初始化錯(cuò)誤，如下圖，
再來看看 這個(gè)SelfAttention 自定義的類的初始化
這就說明再調(diào)用這個(gè)類的時(shí)候，輸入的ch=256并不會初始化這個(gè)類，需要先自定義好初始化值，如下圖
調(diào)用方式不變
這樣問題就解決啦！
補(bǔ)充知識：keras load model的時(shí)候，報(bào)錯(cuò)('Keyword argument not understood:', u'******')如何解決
由于keras不同版本的API有變化，因此在一個(gè)keras版本下訓(xùn)練的模型在另一個(gè)keras版本下加載時(shí)，可能會出現(xiàn)諸如('Keyword argument not understood:', u'data_format')等報(bào)錯(cuò)。
通過打開*.h5文件，查看該模型訓(xùn)練所用環(huán)境，再安裝該環(huán)境即可解決報(bào)錯(cuò)。
查看Keras Model所用的Keras環(huán)境的方法
import h5py
f = h5py.File('Model.h5', 'r')
print(f.attrs.get('keras_version'))
根據(jù)輸出的keras版本安裝對應(yīng)版本的keras即可解決加載問題。
以上這篇keras的load_model實(shí)現(xiàn)加載含有參數(shù)的自定義模型就是小編分享給大家的全部內(nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
相關(guān)文章
這篇文章主要介紹了python中if和elif的區(qū)別，下面文章將舉多個(gè)例子圍繞if和eli的相關(guān)資料展開內(nèi)容,需要的朋友可以參考一下，希望對你有所幫助2021-11-11
Python使用win32com模塊實(shí)現(xiàn)數(shù)據(jù)庫表結(jié)構(gòu)自動生成word表格的方法
這篇文章主要介紹了Python使用win32com模塊實(shí)現(xiàn)數(shù)據(jù)庫表結(jié)構(gòu)自動生成word表格的方法,結(jié)合實(shí)例形式分析了win32com模塊下載、連接mysql、查詢獲取表結(jié)構(gòu)以及使用win32com生成word表格的相關(guān)操作技巧,需要的朋友可以參考下2018-07-07
python3 解決requests出錯(cuò)重試的問題
這篇文章主要介紹了python3 解決requests出錯(cuò)重試的問題，具有很好的參考價(jià)值，希望對大家有所幫助。一起跟隨小編過來看看吧2021-04-04
跟老齊學(xué)Python之玩轉(zhuǎn)字符串(1)
本文主要講訴了字符串的定義，變量鏈接到字符串，以及對字符串的簡單操作，都是些非常基礎(chǔ)的東西，適合零基礎(chǔ)的Pythoner學(xué)習(xí)，覺得有用的話，多鼓鼓掌吧2014-09-09
基于python實(shí)現(xiàn)藍(lán)牙通信代碼實(shí)例
這篇文章主要介紹了基于python實(shí)現(xiàn)藍(lán)牙通信代碼實(shí)例,文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2019-11-11
這篇文章主要介紹了pandas 如何分割字符的實(shí)現(xiàn)方法，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2019-07-07
這篇文章主要介紹了詳解pygame捕獲鍵盤事件的兩種方式，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2021-04-04
使用python創(chuàng)建股票的時(shí)間序列可視化分析
這篇文章主要為大家詳細(xì)介紹了python創(chuàng)建股票的時(shí)間序列可視化分析，文中示例代碼介紹的非常詳細(xì)，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下，希望能夠給你帶來幫助2022-03-03
這篇文章主要介紹了Python partial函數(shù)原理及用法解析,文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2019-12-12
這篇文章主要介紹了Python抓取Discuz!用戶名腳本代碼，有需要的朋友可以參考一下2013-12-12