在服務器上使用 keras 對數據進行訓練完成后,通過 save 的方法將模型保存成對應的 xxxxx.h5 文件,將對應的 h5 文件拷貝到 windows 下面的時候通過 load 命令加載卻報錯。
報錯如下:
keras load_model ValueError: Unknown initializer: GlorotUniform。
這是因為在服務器上安裝 keras 時候是在 tensorflow 里面進行的,而 Windows 下面進行加載的時候導入的卻是 keras 下面的加載模型,把對應的導入模塊進行替換即可。
from keras.models import load_model # 將上面的這句話替換成下面的,即可。 from tensorflow.keras.models import load_model # 在執行此句就不會報錯了 model = load_model('lstm_model.h5')