问题描述:我是直接新建了一个.ipynb文件后,开始跑代码,断开连接后(比如下工了回宿舍睡觉),下次再打开,生成的模型文件并没有存储,又得重新训练一遍。因此,希望把模型文件存储在drive中,下次就能直接用模型预测,不用再跑一遍代码了。
方法一:
新建一个colab文件:
点击装载Google云端硬盘:
出现drive文件夹:
运行一个简单的实例:
成功保存。
方法二(复杂):
运行挂载代码,具体请百度google drive挂载。
!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}
!mkdir -p drive
!google-drive-ocamlfuse drive
from google.colab import drive
drive.mount('/content/drive/')
进入指定目录,因为我的代码是直接新建后跑的,因此与drive都在\content目录下
import os
os.chdir('/content')
我的目录结构如下:
然后执行命令拷贝:
!cp -r checkpoints '/content/drive/MyDrive/ML-2021-spring-yiLee/2020HW9'
这样就把checkpoints这个文件夹保存在drive中,下次打开colab中,文件依然在,可以直接调用。
希望对大家有帮助。