Получить сводку по модели tenorflow - PullRequest
1 голос
/ 20 февраля 2020

Model.summary () дает мне этот вывод model summary

Теперь, как я могу проверить слои sequential_1 и sequential_3? Мне нужна вся сводка по модели, но она дает две последовательные, так что это означает, что две модели объединяются, так как я могу получить сводку по обеим моделям? У меня есть только model.h5 файл, ничего больше

1 Ответ

0 голосов
/ 25 марта 2020

Модели, сохраненные в формате .h5, включают в себя все, что касается модели.

Для проверки сводки слоев внутри модели в модели, как в вашем случае. Вы можете извлечь слои, а затем вызвать метод сводки для каждого из них.

ie. layer_summary = [layer.summary() for layer in loaded_model.layers]


Вот полный код, который я использовал при воспроизведении вашего сценария.

import tensorflow as tf
print('Running Tensorflow version {}'.format(tf.__version__)) # Tensorflow 2.1.0

model_path = '/content/keras_model.h5'

loaded_model = tf.keras.models.load_model(model_path)
loaded_model.summary()

inp = loaded_model.input   
layer_summary = [layer.summary() for layer in loaded_model.layers]   

Я также использовал файл model.h5 , который вы загружено.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...