Как сохранять и загружать модели Deep Learning / Machine Learning в Google Colab с помощью Google Drive

Сохранение обученных моделей на Google Диск

Используйте следующий код для сохранения моделей в Google Colab.

0. Установите библиотеку PyDrive в блокнот Google Colab.

!pip install -U -q PyDrive
from pydrive.auth import GoogleAuth
from pydrive.drive import GoogleDrive 
from google.colab import auth 
from oauth2client.client import GoogleCredentials

1. Аутентифицируйте и создайте клиент PyDrive.

auth.authenticate_user()
gauth = GoogleAuth()
gauth.credentials = GoogleCredentials.get_application_default()                       drive = GoogleDrive(gauth)

2. Сохраните модель или веса на Google Диске и создайте в каталоге Colab на Google Диске.

model.save('model.h5')
model_file = drive.CreateFile({'title' : 'model.h5'})                       model_file.SetContentFile('model.h5')                       model_file.Upload()

3. Скачиваем модель на гугл диск

# download to google drive                       drive.CreateFile({'id': model_file.get('id')})

Загрузка моделей с Google Диска в Colab

file_obj = drive.CreateFile({'id': '16zbkEN4vqnPvIOvLMF1IaXKYCq5g3Yln'})                       file_obj.GetContentFile('keras.h5')