Как загрузить сохраненную в keras модель с помощью пользовательского Оптимизатора - PullRequest
0 голосов
/ 23 февраля 2019

Я скомпилировал и обучил модель keras с помощью специального оптимизатора.Я сохранил модель, но при попытке загрузить модель выдает ошибку, указывающую ValueError: Unknown optimizer: MyOptimizer.Я попытался передать MyOptimizer как пользовательский объект, например, models.load_model('myModel.h5', custom_objects={'optimizer':MyOptimizer}), и он все равно выдает ошибку.Как загрузить модель кераса с пользовательскими объектами?

1 Ответ

0 голосов
/ 12 июня 2019

Я столкнулся с той же проблемой:)

Я заставил это работать, загрузив модель с models.load_model('myModel.h5', compile=False).

Из исходного кода keras :

Если в составе сохраненной модели был найден оптимизатор, модель уже скомпилирована.В противном случае модель будет скомпилирована, и появится предупреждение.Когда для compile установлено значение False, компиляция пропускается без каких-либо предупреждений.

После загрузки некомпилированной модели я могу скомпилировать ее снова с помощью своего пользовательского оптимизатора.

...