Colab保存模型到google drive

一般在colab中训练好模型之后，我们都希望保留模型参数或者实验结果，这样下次就可以直接访问而不需要再次run model。

首先保存模型的代码比较简单，比如用pytorch的话:

import torch
torch.save(model, './model.pt')

或者用pickle保存模型结果，具体怎么使用pickle可以看我的另一篇文章。

但问题是，这个保存并没有和自己的google drive云盘绑定，实际上只是保存在了colab的当前内存中。。等你下次刷新就没了。

所以在存模型之前需要挂到drive上，并且进入drive的路径：

#connect to self drive
from google.colab import drive
drive.mount('/content/drive')

import os
os.chdir('/content/drive/My Drive')

然后你存的任何东西都在drive的首页路径下了，下次读取之前也先跑上面这段代码。如果想更改其他路径，用os.chdir即可。