Colaboratory上训练好的模型怎么保存

Colab是一个能提供免费GPU/TPU算力的一个平台,我们可以使用其算力来加速程序,而每隔12个小时分配给我们的资源会强制收回,就需要我们学会上传、下载、保存我们训练好的模型,这里讲一下怎么保存训练好的模型。

连接Google硬盘

使用Google账号注册一个Google Drive的账号,注册成功就可以获得15G的云存储空间,这时我们只需要将数据和文件上传到Google Drive上,在Colab中就可以直接使用数据和文件

创建一个新的Colab文件,执行一下代码

!apt-get install -y -qq software-properties-common python-software-properties module-init-tools
!add-apt-repository -y ppa:alessandro-strada/ppa 2>&1 > /dev/null
!apt-get update -qq 2>&1 > /dev/null
!apt-get -y install -qq google-drive-ocamlfuse fuse
from google.colab import auth
auth.authenticate_user()
from oauth2client.client import GoogleCredentials
creds = GoogleCredentials.get_application_default()
import getpass
!google-drive-ocamlfuse -headless -id={creds.client_id} -secret={creds.client_secret} < /dev/null 2>&1 | grep URL
vcode = getpass.getpass()
!echo {vcode} | google-drive-ocamlfuse -headless -id={creds.client_id} -secret={creds.client_secret}

运行以上代码后会得到两个连接,点击连接,进行相应的授权后,会得到相应的验证码
Colaboratory上训练好的模型怎么保存_第1张图片
Colaboratory上训练好的模型怎么保存_第2张图片

挂载Google Drive

!mkdir -p drive
!google-drive-ocamlfuse drive

运行结果
Colaboratory上训练好的模型怎么保存_第3张图片

此时云盘里的文件已经和Colab里的文件连接,直接保存就可

举个简单的例子:
我要保存这份命名为model的模型
Colaboratory上训练好的模型怎么保存_第4张图片
运行结果
Colaboratory上训练好的模型怎么保存_第5张图片
后面直接调用就可,这样就不用后面直接调用就可,也不会被收回

你可能感兴趣的:(其他)