如何在Python中保存所有深度学习模型参数?

杜尚·内德列科维奇

是否可以保存在keras中实现的LSTM模型中的所有参数?当我应用以下功能时:

model = load_model("model_RNN.h5")
for layer in model.layers:
    g = layer.get_config()
    weights = layer.get_weights()

仅获取与最后一层相关的参数(17个权重)?我的LSTM网络体系结构如下,包含7505个参数。如何保存所有7505参数?

_________________________________________________________________
Layer (type)                 Output Shape              Param #
=================================================================
lstm (LSTM)                  (None, 10, 32)            4352
_________________________________________________________________
lstm_1 (LSTM)                (None, 16)                3136
_________________________________________________________________
dense (Dense)                (None, 1)                 17
=================================================================
Total params: 7,505
Trainable params: 7,505
Non-trainable params: 0
史努比博士

您所谓的“参数”在Keras中称为“权重”,当然,Model实例具有一种get_weights()方法,您可以使用方法来获取包含所有层的所有权重的列表:

weights = model.get_weights()

然后,您可以使用所需的任何技术保存此列表。

本文收集自互联网,转载请注明来源。

如有侵权,请联系[email protected] 删除。

编辑于
0

我来说两句

0条评论
登录后参与评论

相关文章

来自分类Dev

如何在python中保存网站上的所有图像

来自分类Dev

如何在云端部署深度学习模型?

来自分类Dev

如何在peewee(python)的单个事务中保存所有实例?

来自分类Dev

pytorch-如何从DistributedDataParallel学习中保存和加载模型

来自分类Dev

如何在联邦的Tensorflow中保存模型

来自分类Dev

如何在PyTorch中保存模型架构?

来自分类Dev

如何在Django中保存模型记录

来自分类Dev

如何在深度学习模型中训练不同尺度的特征

来自分类Dev

如何在Accord.net中保存“学习状态”?

来自分类Dev

如何在Ember.js中保存具有多对多关系的模型?

来自分类Dev

如何在Vapor 3中保存具有特定ID的模型

来自分类Dev

如何在Ember.js中保存具有多对多关系的模型?

来自分类Dev

人脸识别深度学习有哪些模型?

来自分类Dev

如何在VBA for Excel中保存所有隐藏列的范围

来自分类Dev

如何在所有 shell 中保存命令历史记录

来自分类Dev

如何在Lambda,Python中保存索引

来自分类Dev

如何在python(pandas)中保存数据

来自分类Dev

如何在python中保存列表?

来自分类Dev

如何在python中保存ping输出

来自分类Dev

如何在一次交易中保存相关模型?

来自分类Dev

如何在Ember Data中保存关联模型上的更改

来自分类Dev

如何在Django模型中保存对象列表?

来自分类Dev

如何在R中保存JAGS模型对象?

来自分类Dev

如何在Scala中保存RandomForestClassifier Spark模型?

来自分类Dev

如何在Apache Spark中保存和加载MLLib模型?

来自分类Dev

如何在一次交易中保存相关模型?

来自分类Dev

如何在libsvm中保存经过matlab训练的模型

来自分类Dev

如何在R中保存JAGS模型对象?

来自分类Dev

如何在视图中的模型中保存表单字段

Related 相关文章

热门标签

归档