やりたいこと
- Google Colabで学習させたモデルを、Googleドライブに保存する
手順
- Google Driveを利用できるようにする
- 以下を実行するとログイン画面に移動。ログインすると表示されるトークンを、入力するボックス(出てくる)に入力する
from google.colab import drive
drive.mount('/content/drive')
- モデルを保存する
- 右クリックで開かれるメニューからパスを取得できて便利
model.save("drive/My Drive/Colab Notebooks/sample/sample.h5")
保存したモデルを使用する
from tensorflow.keras.models import load_model
saved_model = load_model('/content/drive/My Drive/Colab Notebooks/sample/sample.h5')