Я использую model.save("cnn.model") и model.save("cnn.h5") для сохранения модели после тренировки.
model.save("cnn.model")
model.save("cnn.h5")
В чем разница сохранения модели в 2 разных расширениях?
Имя файла, которое включает расширение, не имеет значения. Что бы это ни было, Keras сохранит модель в формате HDF5 в этот файл.
Do c: Как сохранить модель Keras?
Вы можно использовать model.save(filepath) для сохранения модели Keras в один файл HDF5, который будет содержать: архитектура модели, позволяющая воссоздать модель весов Модель Конфигурация обучения (потери, оптимизатор) Состояние оптимизатора, позволяющее возобновить обучение именно с того места, где вы остановились.
Вы можно использовать model.save(filepath) для сохранения модели Keras в один файл HDF5, который будет содержать:
model.save(filepath)