我想获取某些层的权重 - 不是全部，因为架构不同 - 来自model_trained并初始化model_untrained用它。我怎样才能用 Keras 做到这一点？
如果你有一个函数create_model()它返回一个 Keras 模型（example https://github.com/fchollet/keras/blob/master/examples/mnist_cnn.py#L47-L57），你可以像这样初始化它的权重：
from keras.models import load_model
model_untrained = create_model()
model_trained = load_model('trained_model.h5')
extracted_weights = model_trained.layers[0].get_weights()
model_untrained.layers[0].set_weights(extracted_weights)
本文内容由网友自发贡献，版权归原作者所有，本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容，请联系:hwhale#tublm.com(使用前将#替换为@)
