keras的load_model實現(xiàn)加載含有參數(shù)的自定義模型
網(wǎng)上的教程大多數(shù)是教大家如何加載自定義模型和函數(shù)，如下圖
這個SelfAttention層是在訓練過程自己定義的一個class，但如果要加載這個自定義層，需要在load_model里添加custom_objects字典，這個自定義的類，不要用import ，最好是直接復制進再訓練的模型中，這些是基本教程。
------------------分割線講重點------------------
如果直接運行上面的代碼，會出現(xiàn)一個init初始化錯誤，如下圖，
再來看看 這個SelfAttention 自定義的類的初始化
這就說明再調(diào)用這個類的時候，輸入的ch=256并不會初始化這個類，需要先自定義好初始化值，如下圖
調(diào)用方式不變
這樣問題就解決啦！
補充知識：keras load model的時候，報錯('Keyword argument not understood:', u'******')如何解決
由于keras不同版本的API有變化，因此在一個keras版本下訓練的模型在另一個keras版本下加載時，可能會出現(xiàn)諸如('Keyword argument not understood:', u'data_format')等報錯。
通過打開*.h5文件，查看該模型訓練所用環(huán)境，再安裝該環(huán)境即可解決報錯。
查看Keras Model所用的Keras環(huán)境的方法
import h5py
f = h5py.File('Model.h5', 'r')
print(f.attrs.get('keras_version'))
根據(jù)輸出的keras版本安裝對應版本的keras即可解決加載問題。
以上這篇keras的load_model實現(xiàn)加載含有參數(shù)的自定義模型就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關文章
這篇文章主要為大家介紹了Pandas字符串操作的各種方法及速度測試,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進步,早日升職加薪2023-08-08
Python實現(xiàn)查找數(shù)據(jù)庫最接近的數(shù)據(jù)
這篇文章主要介紹了Python實現(xiàn)查找數(shù)據(jù)庫最接近的數(shù)據(jù),文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下2020-06-06
Python cookbook（數(shù)據(jù)結構與算法）從任意長度的可迭代對象中分解元素操作示例
這篇文章主要介紹了Python 數(shù)據(jù)結構與算法 從任意長度的可迭代象中分解元素操作,結合實例形式分析了Python使用*表達式針對可迭代對象的分解操作相關實現(xiàn)技巧,需要的朋友可以參考下2018-02-02
使用?Flask、Celery?和?Python?實現(xiàn)每月定時任務的步驟
下面給大家分享使用?Flask、Celery?和?Python?實現(xiàn)每月定時任務的步驟,本文分步驟結合腳本給大家介紹的非常詳細,感興趣的朋友跟隨小編一起看看吧2024-08-08
這是去年畢設做的一個Web漏洞掃描小工具，主要針對簡單的SQL注入漏洞、SQL盲注和XSS漏洞。下文給大家介紹了使用說明和源代碼，一起看看吧2016-10-10
Python CategoricalDtype自定義排序實現(xiàn)原理解析
這篇文章主要介紹了Python CategoricalDtype自定義排序實現(xiàn)原理解析,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下2020-09-09
用Python實現(xiàn)一個打字速度測試工具來測試你的手速
有很多小伙伴們都苦惱自己手速不夠,今天特地整理了這篇文章,教你用Python實現(xiàn)一個打字測試工具來測試你的打字速度,文中有非常詳細的代碼示例,對想練手速的小伙伴們很有用哦,需要的朋友可以參考下2021-05-05
數(shù)據(jù)清洗是數(shù)據(jù)處理過程中至關重要的一部分,本文主要介紹了如何使用Python數(shù)據(jù)清洗庫,具有一定的參考價值,感興趣的可以了解一下2023-12-12
這篇文章主要介紹了自定義PyCharm快捷鍵的設置方式，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2023-05-05
python+pygame實現(xiàn)簡易五子棋小游戲的三種方式
這篇文章主要介紹了使用python實現(xiàn)簡易五子棋小游戲,文中提供了三種實現(xiàn)方式,解決思路和部分實現(xiàn)代碼,感興趣的朋友可以參考下2023-03-03