如何僅爲某些圖層加載權重？

Question

我想採取一些層的權重 - 不是所有的，因爲體系結構不同 - 從model_trained和初始model_untrained與它。我怎麼能用Keras做到這一點？

Answer 1

如果你有一個函數create_model()返回一個Keras模型（example），可以初始化這樣的權重：

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)