Как НЕ сохранить оптимизатор модели в Tensorflow Keras?

Question

Я читаю официальный учебник по сохранению-загрузке в Keras, и кажется, что я использовал save или save_weights методы, тогда параметры оптимизатора будут сохранены в любом случае. Как сохранить только вес модели?

Answer 1

В Keras, чтобы сохранить веса моделей, выполните:

model.save_weights('my_model_weights.h5')

Чтобы загрузить веса моделей:

model.load_weights('my_model_weights.h5')

Также см. Дополнительный пример сохранения / загрузки весов по имени слоя из здесь .