save_model () и load_model () в tenorflow.keras - PullRequest
0 голосов
/ 09 апреля 2020

Я использую keras save_model() и .load_model(), чтобы сохранить и загрузить эту сохраненную модель.

Я сохраняю ее с помощью save_model() и затем загрузите ранее сохраненный файл в новый, используя load_model(save_path):

import tensoflow as tf

save_path='/home/username/my_model.hf5'
saved_model = tf.keras.models.save_model(model, 
                                         save_path, 
                                         include_optimizer = True,
                                         save_format = 'h5')
new_model = tf.keras.models.load_model(save_path)

Но при использовании load_model() возникает ошибка при загрузке состояния оптимизатора;

WARNING:tensorflow:Error in loading the saved optimizer state.    
As a result, your model is starting with a freshly initialized optimizer.

, так что Недавно инициализированный оптимизатор запущен, хотя я установил include_optimizer = True в save_model().

Значит ли это, что нет весов? И как я могу загрузить всю сохраненную модель с тренированными весами?

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