Как получить доступ и визуализировать веса в предварительно обученной модели TensorFlow 2?

Question

Итак, я заново обучил предварительно обученную модель ResNet50 V2 в TensorFlow 2, используя среду tf.keras с двумя плотными слоями, добавленными сверху. Теперь я хочу визуализировать веса в слоях внутри базовой модели Re sNet. Однако перезагрузка сохраненной модели с

model = tf.keras.models.load_model(path/to/model.hdf5)
model.summary()

приводит к

Как видите, слои Re sNet модель не указана отдельно, это означает, что вызов

model.layers[0].get_weights()[1]

приведет только к

[7 7 3 64]

Таким образом, как мне получить доступ к весам внутри каждого из слоев в базовом ResNet50 V2? модель?

Answer 1

Правильный ответ здесь был написать

model.layers[0].summary()

вместо

model.summary()

, что позволит мне тогда увидеть, что все слои находятся в предварительно обученной модели. Таким образом, написание

model.layers[0].layers[0].get_weights(name='input_1')

даст мне вес входных данных для базовой модели Re sNet.