Загружать веса керас до определенного слоя

Question

Я тренировал 3dCNN модель.Я хотел бы загрузить свои сохраненные веса в Керас до определенного слоя.Я назвал определенный слой как name = "my_dense_layer".Как я могу загрузить веса до этого слоя?

Моя модель выглядит так:

Я хочу загрузить свои веса попропуская последний выпадающий слой и плотный слой с размером 6. Затем я хотел бы спроецировать свой ввод размером 32x10x28x28 и использовать вес конкретного плотного слоя с выходным размером вектора 512 измерений.Таким образом, иметь в качестве вывода 32x512.

Answer 1

Поскольку сохраненные веса расположены как b1, w1, b2, w2 ..., мы можем загружать веса слой за слоем до my_dense_layer. Возможное решение следующим образом:

for layer in model.layers:
if layer.trainable_weights:
    b=weights.pop(0)
    w=weights.pop(0)
    layer.set_weights((b,w))
    if layer.name=='my_dense_layer':
        break