Keras中加入Lambda層無法正常載入模型問題

使用Lambda層加入自定義的函數(shù)后,訓(xùn)練沒有出錯(cuò),但是在載入保存的模型時(shí)顯示錯(cuò)誤。

1."AttributeError: 'NoneType' object has no attribute 'get'"

解決方法:這個(gè)問題是由于缺少config信息導(dǎo)致的。Lambda層在載入時(shí)需要載入一個(gè)函數(shù),當(dāng)使用自定義函數(shù)時(shí),模型無法找到則個(gè)函數(shù),也就構(gòu)建不了模型。

load_model(path, custom_objects={"batch_d":self.batch_d, "trans":self.trans})

2."name: 'tf' is not defined"

解決方法:在config信息里面加一條 custom_objects={'tf': tf}

load_model(path, custom_objects={"batch_d":self.batch_d, "trans":self.trans, 'tf': tf})

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡書系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

友情鏈接更多精彩內(nèi)容