いくつかのレイヤーでのみウェイトを読み込むことはできますか？

Question

私はいくつかのレイヤーの重みを取っています - すべてがアーキテクチャと異なるので、model_trainedからそれを取り除き、model_untrainedを初期化します。どのようにKerasでこれを行うことができますか？

Answer 1

あなたはKerasモデル（example）を返す関数create_model()を持っている場合、あなたはこのようにその重みを初期化することができます。

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)