Как преобразовать КПП в модель Keras .h5? - PullRequest
0 голосов
/ 25 мая 2020

У меня есть модель тензорного потока, которая сохраняет контрольные точки, но мне нужно загрузить веса и сохранить модель Kereas .h5. Как я могу это сделать?

1 Ответ

0 голосов
/ 25 мая 2020

Я предполагаю, что вам нужно преобразовать вашу предыдущую контрольную точку в .h5

Учитывая уже обученную модель, вы хотите загрузить ее веса и сохранить как .h5 . Я предполагаю, что вы сохранили его как файл .model . Допустим, он назывался first.model

В вашем скрипте вы захотите использовать load_model, загрузив контрольную точку с помощью

model = load_model('first.model')

, тогда вам просто нужно будет используйте

model.save('goal.h5')

, чтобы сохранить как файл .h5.

Для дальнейшего использования вы можете избежать этого процесса преобразования, сохранив контрольные точки как .h5:

При использовании Функция контрольных точек, у вас есть возможность сохранить как .model .h5 или .hdf5 . Строка может выглядеть примерно так:

checkpoint = ModelCheckpoint("**FILE_NAME_HERE**.model",monitor='val_loss',verbose=1,mode='min',save_best_only=True,save_weights_only=False,period=1)

Таким образом вы сохраняете контрольную точку как .model, но чтобы сохранить ее как h5, как вы и хотите:

checkpoint = ModelCheckpoint("**FILE_NAME_HERE**.h5",monitor='val_loss',verbose=1,mode='min',save_best_only=True,save_weights_only=False,period=1)
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...