Как я должен развернуть обученную модель из Google Colab на моем компьютере

Question

Я использую Google Colab Laboratory для обучения моей модели через набор данных, который я загрузил в colab.research.google.com.После завершения процесса обучения я хочу использовать обученные параметры на своем локальном ПК и создать ту же модель с обученными значениями локально.Какой процедуре мне следует следовать?

Answer 1

Вам необходимо подключить свой ноутбук к своему накопителю Google с помощью:

from google.colab import drive
drive.mount('/content/gdrive')

, затем после обучения сохраните обученную модель как:

model.save("/content/gdrive/My Drive/name_of_your_model.h5")

Откройте накопитель Google, загрузитесохраненный файл name_of_your_model.h5.Затем на свой локальный компьютер загрузите модель

from keras.models import load_model
model = load_model(<path to your model file on local machine>)

Теперь у вас есть обученная модель на вашем локальном компьютере