В чем разница при сохранении модели как cnn.model или cnn.h5? Чем отличаются эти расширения? - PullRequest
0 голосов
/ 26 января 2020

Я использую model.save("cnn.model") и model.save("cnn.h5") для сохранения модели после тренировки.

В чем разница сохранения модели в 2 разных расширениях?

1 Ответ

2 голосов
/ 26 января 2020

Имя файла, которое включает расширение, не имеет значения. Что бы это ни было, Keras сохранит модель в формате HDF5 в этот файл.

Do c: Как сохранить модель Keras?

Вы можно использовать model.save(filepath) для сохранения модели Keras в один файл HDF5, который будет содержать:

  • архитектура модели, позволяющая воссоздать модель
  • весов Модель
  • Конфигурация обучения (потери, оптимизатор)
  • Состояние оптимизатора, позволяющее возобновить обучение именно с того места, где вы остановились.
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...