Коллаборация Google, Keras: сохраните модель в формате HDF5 и загрузите ее на ноутбук - PullRequest
Новая
       42

Коллаборация Google, Keras: сохраните модель в формате HDF5 и загрузите ее на ноутбук

0 голосов
/ 21 мая 2018

Я тренирую маленькую модель 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

1 Ответ

0 голосов
/ 05 августа 2018

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

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

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

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

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

from google import files
files.upload("model1.h5")

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

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

...