溫馨提示×
您好，登錄后才能下訂單哦！
點擊 登錄注冊 即表示同意《億速云用戶服務條款》
您好，登錄后才能下訂單哦！
本篇文章展示了load model實現加載含有參數自定義類的方法具體操作，代碼簡明扼要容易理解，絕對能讓你眼前一亮，通過這篇文章的詳細介紹希望你能有所收獲。
Keras在代碼結構上由面向對象方法編寫，完全模塊化并具有可擴展性，其運行機制和說明文檔有將用戶體驗和使用難度納入考慮，并試圖簡化復雜算法的實現難度 。
加載自定義模型和函數，如下圖
這個SelfAttention層是在訓練過程自己定義的一個class，但如果要加載這個自定義層，需要在load_model里添加custom_objects字典，這個自定義的類，不要用import ，最好是直接復制進再訓練的模型中。
如果直接運行上面的代碼，會出現一個init初始化錯誤，如下圖，
再來看看 這個SelfAttention 自定義的類的初始化
這就說明再調用這個類的時候，輸入的ch=256并不會初始化這個類，需要先自定義好初始化值，如下圖
調用方式不變
這樣問題就解決啦！
補充知識：keras load model的時候，報錯('Keyword argument not understood:', u'******')如何解決
由于keras不同版本的API有變化，因此在一個keras版本下訓練的模型在另一個keras版本下加載時，可能會出現諸如('Keyword argument not understood:', u'data_format')等報錯。
通過打開*.h6文件，查看該模型訓練所用環境，再安裝該環境即可解決報錯。
查看Keras Model所用的Keras環境的方法
import h6py
f = h6py.File('Model.h6', 'r')
print(f.attrs.get('keras_version'))根據輸出的keras版本安裝對應版本的keras即可解決加載問題。
看完上述內容，你們掌握load model實現加載含有參數自定義類的方法了嗎？如果還想學到更多技能或想了解更多相關內容，歡迎關注億速云行業資訊頻道，感謝各位的閱讀！
免責聲明：本站發布的內容（圖片、視頻和文字）以原創、轉載和分享為主，文章觀點不代表本網站立場，如果涉及侵權請聯系站長郵箱：is@yisu.com進行舉報，并提供相關證據，一經查實，將立刻刪除涉嫌侵權內容。
