如何仅加载某些层的权重？

Question

我想取一些层的权重 - 不是全部，因为架构不同 -model_trained并model_untrained用它初始化。我怎样才能用 Keras 做到这一点？

Answer 1

如果您有一个create_model()返回 Keras 模型的函数（示例），您可以像这样初始化它的权重：

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)