溫馨提示×
您好，登錄后才能下訂單哦！
點擊 登錄注冊 即表示同意《億速云用戶服務條款》
您好，登錄后才能下訂單哦！
這篇文章給大家分享的是有關Tensorflow2.0 tf.keras.Model.load_weights() 報錯的處理方法的內容。小編覺得挺實用的，因此分享給大家做個參考。一起跟隨小編過來看看吧。
錯誤描述：
1、保存模型：model.save_weights('./model.h5')
2、腳本重啟
3、加載模型：model.load_weights('./model.h5')
4、模型報錯：ValueError: You are trying to load a weight file containing 12 layers into a model with 0 layers.
問題分析：
模型創建后還沒有編譯，一般是在模型加載前調用model.build(input_shape), 但我通過Dataset將輸入已經變為dict格式了，暫時沒找這樣輸入怎么匹配input_shape參數
解決方法：
model.fit(train_dataset, epochs=0)
將epochs設為0，這樣模型在編譯的同時不會訓練數據，減少耗費的時間，之后就可以正常加載保存的參數了。
感謝各位的閱讀！關于Tensorflow2.0 tf.keras.Model.load_weights() 報錯的處理方法就分享到這里了，希望以上內容可以對大家有一定的幫助，讓大家可以學到更多知識。如果覺得文章不錯，可以把它分享出去讓更多的人看到吧！
免責聲明：本站發布的內容（圖片、視頻和文字）以原創、轉載和分享為主，文章觀點不代表本網站立場，如果涉及侵權請聯系站長郵箱：is@yisu.com進行舉報，并提供相關證據，一經查實，將立刻刪除涉嫌侵權內容。
