有时候我们保存模型（save model）,会保存整个模型输入到输出的权重，如果，我们不想保存后几层的参数，保存成新的模型。

    import kerasfrom keras.models import Model, load_modelfrom keras.layers import Input, Densefrom keras.optimizers import RMSpropimport numpy as np

创建原始模型并保存权重

inputs = Input((1,))
dense_1 = Dense(10, activation='relu')(inputs)
dense_2 = Dense(10, activation='relu')(dense_1)
dense_3 = Dense(10, activation='relu')(dense_2)
outputs = Dense(10)(dense_3)model = Model(inputs=inputs, outputs=outputs)
model.compile(optimizer=RMSprop(), loss='mse')
model.save('test.h5')

加载模型并对模型进行调整

loaded_model = load_model('test.h5')
loaded_model.layers.pop()
loaded_model.layers.pop()

此处去掉了最后两层－－dense_3, dense_2。

创建新的model并加载修改后的模型

new_model = Model(inputs=inputs, outputs=dense_1)
new_model.compile(optimizer=RMSprop(), loss='mse')
new_model.set_weights(loaded_model.get_weights())new_model.summary()
new_model.save('test_complete.h5')

参考：https://github.com/keras-team/keras/issues/8772

keras load model 并保存特定层 (pop) 的权重save new_model相关推荐

keras的model保存和载入
保存keras的model文件和载入keras文件的方法有很多. keras中的模型主要包括model和weight两个部分. 保存模型结构保存model部分的主要方法:一是通过json文件,二是通 ...
TensorFlow中的Keras用法和自定义模型和层
Keras Keras 是一个用于构建和训练深度学习模型的高阶 API.它可用于快速设计原型.高级研究和生产,具有以下三个主要优势: 方便用户使用 Keras 具有针对常见用例做出优化的简单而一致的界 ...
Keras框架下的保存模型和加载模型
在Keras框架下训练深度学习模型时,一般思路是在训练环境下训练出模型,然后拿训练好的模型(即保存模型相应信息的文件)到生产环境下去部署.在训练过程中我们可能会遇到以下情况: 需要运行很长时间的程序在 ...
keras中的模型保存和加载
tensorflow中的模型常常是protobuf格式,这种格式既可以是二进制也可以是文本.keras模型保存和加载与tensorflow不同,keras中的模型保存和加载往往是保存成hdf5格式. ...
Tensorflow学习之tf.keras(一) tf.keras.layers.Model(另附compile，fit)
模型将层分组为具有训练和推理特征的对象. 继承自:Layer, Module tf.keras.Model(*args, **kwargs ) 参数 inputs 模型的输入:keras.Input ...
Keras调用model.load_weights时报错
报错:ValueError: Unable to load weights saved in HDF5 format into a subclassed Model which has not cre ...
keras神经网络模型的保存与加载
一旦你利用Keras完成了训练,你可以将你的网络保存在HDF5里面. keras的模型保存分为多种情况. 一.不保存模型只显示大概结构 model.summary() 这个函数会打印模型结构,但是仅仅 ...
“让Keras更酷一些！”：层与模型的重用技巧
作者丨苏剑林单位丨追一科技研究方向丨NLP,神经网络个人主页丨kexue.fm 今天我们继续来深挖 Keras,再次体验 Keras 那无与伦比的优雅设计.这一次我们的焦点是"重用&q ...
使用gensim.models.Word2Vec.load(‘model.txt‘)报错，导致模型加载不了的解决办法之一
背景: 想做一个基于Word2Vec的分析标题与标题之间相关性的模型,训练完之后保存模型.再加载发生了如下错误: 在这里插入代码片 Traceback (most recent call last): ...
Wire Load Model (线负载模型)
--------------------- --------------------- --------------------- --------------------- ------------ ...

keras load model 并保存特定层 (pop) 的权重save new_model

创建原始模型并保存权重

加载模型并对模型进行调整

创建新的model并加载修改后的模型

keras load model 并保存特定层 (pop) 的权重save new_model相关推荐

最新文章

热门文章