Я использую 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()
.
Значит ли это, что нет весов? И как я могу загрузить всю сохраненную модель с тренированными весами?