Сотрудничество Google, Керас: Сохраните модель в формате файла HDF5 и загрузите ее на ноутбук.

Я обучаю небольшую модель RNN в Google Collab с использованием графического процессора. Обычно я сохраняю свою модель и веса в формате файла HDF5.

На локальном компьютере (ноутбуке) я делаю это в следующей процедуре

sudo pip install h5py
model.fit(....)
model.save('model1.h5')

Я загружаю обученную модель, чтобы сделать прогноз, используя,

from keras.models import load_model
model = load_model('model1.h5')

Я сейчас хочу

  1. сохранить модель в Google Collab, аналогичный формату, указанному выше
  2. загрузить файл .h5 на локальный компьютер (ПК)
  3. делать прогнозы на ПК и тренировать их на ПК
  4. сохранить обученную модель, загрузить файл .h5 обратно в Google Collab
  5. возобновить обучение в Google Collab

keras h5py google-colaboratory hdf5
person ankushbraj    schedule 21.05.2018    source источник

Ответы (1)


arrow_upward
3
arrow_downward

Немного поздно, но ради будущих разработчиков постараюсь решить вопрос.

Процедура такая же, как на вашем локальном компьютере, с двумя отличиями:

Чтобы загрузить модель из Google Collab:

from google.colab import files
files.download("model1.h5")

Чтобы загрузить модель в Google Collab:

from google.colab import files
files.upload()

Вы можете проверить этот блокнот, чтобы узнать о других вариантах ввода-вывода: https://colab.research.google.com/notebooks/io.ipynb

Все остальные шаги выполняются так же, как и на локальном компьютере. Надеюсь, это поможет тебе.

person Alberto Rincon    schedule 05.08.2018
comment
Исправление from google.colab import files Кроме того, files.upload () не принимает никаких аргументов, поэтому это не сработает. - person ic10503; 04.03.2020