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