項目過程中遇到的問題

1、VAE由於loss函數是自己定義的,在load_model別人訓練好的模型的時候

vae = load_model('E:\AItask......'.format(ftype, alertclass, model_filepath), custom_objects={'vae_loss': vae_loss})

報錯‘tuple index out of range’,後續又改爲報錯‘退出代碼-1073741819 (0xC0000005)’

 

解決:本以爲是傳參的問題,後來改爲在自己電腦上重新訓練了一下,再加載模型就正常了,再後來查了一下可能是因爲python版本問題。問題暫時解決(仍存在問題,在其他不同python版本跑出來的AE模型加載是正常的)。

https://github.com/keras-team/keras/issues/8079

發佈了16 篇原創文章 · 獲贊 8 · 訪問量 4萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章