Colab保存模型到google drive

FrenchOldDriver

于 2021-06-07 13:45:06 发布

阅读量7.9k

点赞数 7

分类专栏：技术文章文章标签： python 机器学习 pytorch

本文链接：https://blog.youkuaiyun.com/OldDriver1995/article/details/117658850

版权

技术文章专栏收录该内容

28 篇文章

订阅专栏

这篇博客介绍了如何在Google Colab中训练模型后将其保存到Google Drive，确保数据不会丢失。首先，通过`google.colab import drive`挂载个人的Google Drive，然后改变工作目录到指定的Drive路径。接着，可以使用`torch.save`或`pickle`保存模型和实验结果。保存的文件将会存储在Drive的指定路径下，下次读取前需再次挂载Drive。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

一般在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即可。