Как сохранить полную модель тензорного потока из Google Colab

Question

Я обучил модель компьютерного зрения в google colab.

Теперь я хочу загрузить ее на свой компьютер, чтобы с ней можно было работать.
Если я сделаю model.save("name.h5"), а затем загрузю файл .h5, сохранит ли это обученную модель?
Или это просто необученная конструкция модели?

Answer 1

Да, mode.save("name.h5") сохраняет обученную модель. Конечно, вы должны выполнить эту строку после того, как вы обучили / подогнали model.

Однако model.save() сохраняет только структуру модели и обновленные веса. И он не хранит весов функций потерь и информацию о функции потерь. Следовательно, вам следует избегать повторного обучения вашей модели после загрузки ее из сохраненного файла.

Если вы хотите продолжить повторное обучение вашей модели, вы должны использовать tf.keras.models.save_model() для сохранения модели и tf.keras.models.load_model() загрузить модель. Дополнительную информацию можно найти в документации Keras . Спасибо.