keras ValueError:Layer“lstm_cell_2”需要3个变量,但在加载期间收到0个变量

pn9klfpd  于 2023-05-18  发布在  其他
关注(0)|答案(1)|浏览(296)

我正在使用Keras将我的Bi-Lstm模型保存在Google Colab上,代码如下

with open('BiLstm_model.pkl', 'wb') as writer:
    pickle.dump(lstm_model, writer)

现在,当我在本地PC中使用代码加载此模型时,它显示了图像中显示的错误。

lstm_model = pickle.load(open('BiLstm_model.pkl', 'rb'))

Error Message
每当我在Google colab中加载模型时,它都在工作,但是当我在本地PC中加载它时,它会给我错误。我已经尝试使用不同的格式和库来保存模型并加载它。此外,我已经安装了相同版本的包指定和使用的谷歌可乐。

dy1byipe

dy1byipe1#

解决方案是尝试使用hdf5保存和加载模型。下面是代码片段

lstm_model.save('BiLstm_model.h5')

并使用

from tensorflow.keras.models import load_model
lstm_model = load_model('./models/BiLstm_model.h5')

相关问题