Керас, как сохранить указанную переменную? - PullRequest
0 голосов
/ 29 июня 2018

Я хочу сохранить свою обученную модель в keras с 10 кратным резюме, и я использую слово встраивание в мою модель, поэтому keras сохраненный файл модели слишком большой, около 550M. а 10-ти кратная обучаемая модель составляет около 5Гб.

Если бы я мог удалить переменную внедрения в сохраненной модели и просто сохранить другую переменную, я думаю, что я мог бы сохранить большую часть размера файла, потому что файл массива внедрения слова составляет около 500M. И тогда общий размер файла будет уменьшен до 1 ГБ.

Но я не могу сделать это с keras.

Также, когда я пытаюсь использовать tf.train.saver в keras, происходит что-то странное.

У кого-нибудь есть идеи?

1 Ответ

0 голосов
/ 29 июня 2018

Вы пытались только сохранить веса вместо всей модели, как указано здесь: https://keras.io/getting-started/faq/#how-can-i-save-a-keras-model

model.save_weights('my_model_weights.h5')
...